Commit 61a13666 authored by 赵威's avatar 赵威

update cpc

parent cc666dfc
......@@ -72,7 +72,7 @@ def user_portrait_increment_scan_info(device_dict, diary_dict, tractate_dict, di
def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn, size=300):
time_begin = time.time()
cpc_set = set(get_unread_cpc_diary_list(device_id, size=size * 10))
cpc_set = set(get_unread_cpc_diary_list(device_id, size=5000))
diary_set_ids = set(user_portrait_tag3_get_candidate_unread_list(device_id, "diary", size=size))
content_diary_ids = list(diary_set_ids.difference(cpc_set))
......@@ -150,21 +150,21 @@ def main():
if __name__ == "__main__":
time_begin = time.time()
# main()
main()
device_dict = device_fe.get_device_dict_from_redis()
diary_dict = diary_fe.get_diary_dict_from_redis()
tractate_dict = tractate_fe.get_tractate_dict_from_redis()
print("redis data: " + str(len(device_dict)) + " " + str(len(diary_dict)) + " " + str(len(tractate_dict)))
# device_dict = device_fe.get_device_dict_from_redis()
# diary_dict = diary_fe.get_diary_dict_from_redis()
# tractate_dict = tractate_fe.get_tractate_dict_from_redis()
# print("redis data: " + str(len(device_dict)) + " " + str(len(diary_dict)) + " " + str(len(tractate_dict)))
diary_save_path = get_essm_model_save_path("diary")
if not diary_save_path:
diary_save_path = "/data/files/models/diary/1597390452"
print(diary_save_path + "!!!!!!!!!!!!!!!!!!!!!!!!!!!")
diary_predict_fn = tf.contrib.predictor.from_saved_model(diary_save_path)
# diary_save_path = get_essm_model_save_path("diary")
# if not diary_save_path:
# diary_save_path = "/data/files/models/diary/1597390452"
# print(diary_save_path + "!!!!!!!!!!!!!!!!!!!!!!!!!!!")
# diary_predict_fn = tf.contrib.predictor.from_saved_model(diary_save_path)
device_id = "androidid_a25a1129c0b38f7b"
offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=300)
# device_id = "androidid_a25a1129c0b38f7b"
# offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=300)
total_time = (time.time() - time_begin) / 60
print("total cost {:.2f} mins at {}".format(total_time, datetime.datetime.now()))
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment