Commit a2224a62 authored by 张彦钊's avatar 张彦钊

add pridict param

parent bb15647c
......@@ -15,6 +15,7 @@ def train():
print("predicting")
ffm_model.setTest(DIRECTORY_PATH + "test_ffm_data.csv")
ffm_model.setSigmoid()
ffm_model.predict(DIRECTORY_PATH + "model.out",DIRECTORY_PATH + "test_set_predict_output.txt")
param_predict = {"log": "/data2/models/result"}
ffm_model.predict(param_predict,DIRECTORY_PATH + "model.out",DIRECTORY_PATH + "test_set_predict_output.txt")
......@@ -17,8 +17,9 @@ def get_date():
day = now.day
date = datetime(year,month,day)
data_start_date = (date - timedelta(days=5)).strftime("%Y-%m-%d")
data_end_date = (date - timedelta(days=2)).strftime("%Y-%m-%d")
validation_date = (date - timedelta(days=3)).strftime("%Y-%m-%d")
data_end_date = (date - timedelta(days=1)).strftime("%Y-%m-%d")
validation_date = (date - timedelta(days=2)).strftime("%Y-%m-%d")
# 验证集和测试集的日期必须相差一天,否则切割数据集时会报错
test_date = data_end_date
print("data_start_date,data_end_date,validation_date,test_date:")
print(data_start_date,data_end_date,validation_date,test_date)
......
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