service model 优化

train/train_service.py

@@ -83,7 +83,7 @@ def getTrainColumns(train_columns,data_vocab):
             emb_columns.append(col)
             inputs[feature] = tf.keras.layers.Input(name=feature, shape=(), dtype='string')
 
-        elif feature.endswith("_number"):
+        elif feature.endswith("_number") and feature.startswith("item"):
             col = tf.feature_column.numeric_column(feature)
             number_columns.append(col)
             inputs[feature] = tf.keras.layers.Input(name=feature, shape=(), dtype='float32')
@@ -112,7 +112,7 @@ def train(emb_columns, number_columns, inputs, train_dataset):
 
     # train the model
     print("train start...")
-    model.fit(train_dataset, epochs=5)
+    model.fit(train_dataset, epochs=1)
     print("train end...")
     print("train save...")