Skip to content

F
ffm-baseline
Commit 88de7fdd authored by 张彦钊's avatar 张彦钊
Browse files
Options

add param in predict

parent 6ced4f40
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
predictDiary.py
View file @ 88de7fdd
...@@ -47,8 +47,8 @@ def predict(user_profile): ...@@ -47,8 +47,8 @@ def predict(user_profile):
ffm_model = xl.create_ffm() ffm_model = xl.create_ffm()
ffm_model.setTest(instance_file_path) ffm_model.setTest(instance_file_path)
ffm_model.setSigmoid() ffm_model.setSigmoid()
param = {"log": "/data2/models/result"}
ffm_model.predict(DIRECTORY_PATH + "model.out", ffm_model.predict(param,DIRECTORY_PATH + "model.out",
DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id'])) DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id']))
print("该用户预测结束") print("该用户预测结束")
predict_save_to_local(user_profile, instance) predict_save_to_local(user_profile, instance)
......
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