Skip to content

G
gm_strategy_cvr
Commit 2e712198 authored by 赵威's avatar 赵威
Browse files
Options

write data

parent c6d355aa
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/main_portrait.py
View file @ 2e712198
......@@ -41,7 +41,7 @@ def offline_predict(device_id, device_dict, diary_dict, predict_fn):
print(diary_ids[:10])
res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
print("res: " + str(len(res)) + "\n")
# user_portrait_tag3_write_ctcvr_data(device_id, "diary", res[:500])
user_portrait_tag3_write_ctcvr_data(device_id, "diary", res[:500])
def main():
......@@ -54,10 +54,10 @@ def main():
save_path = "/home/gmuser/data/models/1595501144"
predict_fn = tf.contrib.predictor.from_saved_model(save_path)
# device_id = "androidid_a25a1129c0b38f7b"
# offline_predict(device_id, device_dict, diary_dict, predict_fn)
device_id = "androidid_a25a1129c0b38f7b"
offline_predict(device_id, device_dict, diary_dict, predict_fn)
user_portrait_scan_info(device_dict, diary_dict, predict_fn)
# user_portrait_scan_info(device_dict, diary_dict, predict_fn)
if __name__ == "__main__":
......
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