美购精排模型

e8f0c408 · 郭羽 · 6f0ad086 · e8f0c408
Commit e8f0c408 authored May 31, 2021 by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

train_service.py train/train_service.py +3 -2

No files found.
--- a/train/train_service.py
+++ b/train/train_service.py
@@ -18,10 +18,11 @@ 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"

-version = time.strftime("%Y%m%d", time.localtime())
+version = "v1"
+trainDay = time.strftime("%Y%m%d", time.localtime())
 data_path_train = "/data/files/service_feature_{}_train.csv".format(version)
 data_path_test = "/data/files/service_feature_{}_test.csv".format(version)
-model_file = "../models/service/" + version
+model_file = "../models/service/" + trainDay

 def is_float(s):
    try: