美购精排模型

1e9de0a3 · 郭羽 · 1ad42e0a · 1e9de0a3
Commit 1e9de0a3 authored 4 years ago by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

train.py mlp/train.py +2 -2

No files found.
--- a/mlp/train.py
+++ b/mlp/train.py
@@ -18,9 +18,9 @@ one_hot_columns = ["item_"+c for c in ["service_type","doctor_type","doctor_famo
 # data_path_train = "/Users/zhigangzheng/Desktop/work/guoyu/service_sort/train/part-00000-a61205d1-ad4e-4fa7-895d-ad8db41189e6-c000.csv"
 # data_path_test = "/Users/zhigangzheng/Desktop/work/guoyu/service_sort/train/part-00000-a61205d1-ad4e-4fa7-895d-ad8db41189e6-c000.csv"

-data_path_train = "/data/files/service_feature_train.csv"
-data_path_test = "/data/files/service_feature_test.csv"
 version = "v1"
+data_path_train = "/data/files/service_feature_{}_train.csv".format(version)
+data_path_test = "/data/files/service_feature_{}_test.csv".format(version)
 model_file = "service_mlp_"+version

 #数据字典