Skip to content

S
serviceRec
Commit 11456633 authored by 郭羽's avatar 郭羽
Browse files
Options

美购精排模型

parent e8f0c408
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
train/train_service.py
View file @ 11456633
...@@ -22,7 +22,7 @@ version = "v1" ...@@ -22,7 +22,7 @@ version = "v1"
trainDay = time.strftime("%Y%m%d", time.localtime()) trainDay = time.strftime("%Y%m%d", time.localtime())
data_path_train = "/data/files/service_feature_{}_train.csv".format(version) data_path_train = "/data/files/service_feature_{}_train.csv".format(version)
data_path_test = "/data/files/service_feature_{}_test.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): def is_float(s):
try: try:
...@@ -156,7 +156,7 @@ def train(emb_columns, number_columns, oneHot_columns, inputs, train_dataset): ...@@ -156,7 +156,7 @@ def train(emb_columns, number_columns, oneHot_columns, inputs, train_dataset):
# train the model # train the model
print("train start...") print("train start...")
model.fit(train_dataset, epochs=1) model.fit(train_dataset, epochs=5)
print("train end...") print("train end...")
print("train save...") print("train save...")
......
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