Commit 48a1d60c authored by 宋柯's avatar 宋柯

Merge remote-tracking branch 'origin/master'

parents d29da8c8 c4e46958
......@@ -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')])
......
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