美购精排模型

train/train_service.py

@@ -22,7 +22,7 @@ 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/" + trainDay
+model_file = "/srv/apps/serviceRec/models/service/" + trainDay
 
 def is_float(s):
     try:
@@ -156,7 +156,7 @@ def train(emb_columns, number_columns, oneHot_columns, inputs, train_dataset):
 
     # train the model
     print("train start...")
-    model.fit(train_dataset, epochs=1)
+    model.fit(train_dataset, epochs=5)
     print("train end...")
     print("train save...")