Skip to content

Commit 9964a1f

Browse files
committed
fix metrics
1 parent 66fdb1c commit 9964a1f

File tree

1 file changed

+1
-1
lines changed
  • src/autogluon/bench/frameworks/autokeras

1 file changed

+1
-1
lines changed

src/autogluon/bench/frameworks/autokeras/exec.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,7 @@ def run(
320320
elif train_data.problem_type in ["multiclass", "classification"]:
321321
output_node = ak.ClassificationHead(metrics=["accuracy"])#metrics=[tf.keras.metrics.Accuracy()])
322322
elif train_data.problem_type == "binary":
323-
output_node = ak.ClassificationHead(metrics=["auc"])#[tf.keras.metrics.AUC(curve="ROC")])
323+
output_node = ak.ClassificationHead(metrics=["AUC"])#[tf.keras.metrics.AUC(curve="ROC")])
324324

325325
# Combine the data into a list for the model
326326
train_data_list = [data for data in [image_data_train, tabular_data_train, text_data_train] if data is not None]

0 commit comments

Comments
 (0)