Skip to content

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

特征工程优化

parent e6cb03df
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train/train_service.py
View file @ c4e46958
...@@ -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