Skip to content

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

预测时模型加了sigmoid函数，修改pandas警告，换了一致取列方式

parent 695faeee
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
predictDiary.py
View file @ c2e53a7c
...@@ -52,6 +52,7 @@ def predict(user_profile): ...@@ -52,6 +52,7 @@ def predict(user_profile):
ffm_model = xl.create_ffm() ffm_model = xl.create_ffm()
ffm_model.setTest(user_instance_file_path) ffm_model.setTest(user_instance_file_path)
ffm_model.setSigmoid()
ffm_model.predict(DIRECTORY_PATH + "model_{0}-{1}_lr{2}_lambda{3}.out".format(DATA_START_DATE, ffm_model.predict(DIRECTORY_PATH + "model_{0}-{1}_lr{2}_lambda{3}.out".format(DATA_START_DATE,
DATA_END_DATE, lr, l2_lambda), DATA_END_DATE, lr, l2_lambda),
DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id'])) DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id']))
...@@ -71,7 +72,7 @@ def upload_predict(user_profile, instance): ...@@ -71,7 +72,7 @@ def upload_predict(user_profile, instance):
def wrapper_result(prob, device_id): def wrapper_result(prob, device_id):
prob = prob.head(500) prob = prob.head(500)
prob["url"] = prob["cid"].apply(lambda x: "http://m.igengmei.com/diary_book/" + str(prob[x.index('|') + 1:]) + '/') prob.loc[:,"url"] = prob["cid"].apply(lambda x: "http://m.igengmei.com/diary_book/" + str(prob[x.index('|') + 1:]) + '/')
prob.to_csv(DIRECTORY_PATH + "result/{}_feed".format(device_id),index= False) prob.to_csv(DIRECTORY_PATH + "result/{}_feed".format(device_id),index= False)
......
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