Skip to content

S
serviceRec
Commit 48a1d60c authored by 宋柯's avatar 宋柯
Browse files
Options

Merge remote-tracking branch 'origin/master'

parents d29da8c8 c4e46958
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train/train_service.py
View file @ 48a1d60c
...@@ -132,7 +132,7 @@ def train(emb_columns, number_columns, oneHot_columns, inputs, train_dataset): ...@@ -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 # compile the model, set loss function, optimizer and evaluation metrics
model.compile( model.compile(
loss='mse', loss='binary_crossentropy',
optimizer='adam', optimizer='adam',
metrics=['accuracy', tf.keras.metrics.AUC(curve='ROC'), tf.keras.metrics.AUC(curve='PR')]) 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