We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 66fdb1c commit 9964a1fCopy full SHA for 9964a1f
src/autogluon/bench/frameworks/autokeras/exec.py
@@ -320,7 +320,7 @@ def run(
320
elif train_data.problem_type in ["multiclass", "classification"]:
321
output_node = ak.ClassificationHead(metrics=["accuracy"])#metrics=[tf.keras.metrics.Accuracy()])
322
elif train_data.problem_type == "binary":
323
- output_node = ak.ClassificationHead(metrics=["auc"])#[tf.keras.metrics.AUC(curve="ROC")])
+ output_node = ak.ClassificationHead(metrics=["AUC"])#[tf.keras.metrics.AUC(curve="ROC")])
324
325
# Combine the data into a list for the model
326
train_data_list = [data for data in [image_data_train, tabular_data_train, text_data_train] if data is not None]
0 commit comments