add param in predict

88de7fdd · 张彦钊 · 6ced4f40 · 88de7fdd
Commit 88de7fdd authored 6 years ago by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

predictDiary.py predictDiary.py +2 -2

No files found.
--- a/predictDiary.py
+++ b/predictDiary.py
@@ -47,8 +47,8 @@ def predict(user_profile):
    ffm_model = xl.create_ffm()
    ffm_model.setTest(instance_file_path)
    ffm_model.setSigmoid()
-
-    ffm_model.predict(DIRECTORY_PATH + "model.out",
+    param = {"log": "/data2/models/result"}
+    ffm_model.predict(param,DIRECTORY_PATH + "model.out",
                      DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id']))
    print("该用户预测结束")
    predict_save_to_local(user_profile, instance)