Skip to content

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

predict

parent 2193e8e9
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/main_portrait.py
View file @ d3d2ff25
...@@ -38,7 +38,7 @@ def offline_predict(device_id, device_dict, diary_dict, predict_fn): ...@@ -38,7 +38,7 @@ def offline_predict(device_id, device_dict, diary_dict, predict_fn):
print("diary_ids") print("diary_ids")
print(diary_ids[:10]) print(diary_ids[:10])
res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn) res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
print(res[:10]) print("res: " + str(len(res)))
# user_portrait_tag3_write_ctcvr_data(device_id, "diary", res) # user_portrait_tag3_write_ctcvr_data(device_id, "diary", res)
...@@ -52,10 +52,10 @@ def main(): ...@@ -52,10 +52,10 @@ def main():
save_path = "/home/gmuser/data/models/1595501144" save_path = "/home/gmuser/data/models/1595501144"
predict_fn = tf.contrib.predictor.from_saved_model(save_path) predict_fn = tf.contrib.predictor.from_saved_model(save_path)
device_id = "androidid_a25a1129c0b38f7b" # device_id = "androidid_a25a1129c0b38f7b"
offline_predict(device_id, device_dict, diary_dict, predict_fn) # 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__": 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