特征工程优化

train/train_service.py

@@ -132,7 +132,7 @@ def train(emb_columns, number_columns, oneHot_columns, inputs, train_dataset):
 
     # compile the model, set loss function, optimizer and evaluation metrics
     model.compile(
-        loss='mse',
+        loss='binary_crossentropy',
         optimizer='adam',
         metrics=['accuracy', tf.keras.metrics.AUC(curve='ROC'), tf.keras.metrics.AUC(curve='PR')])