Skip to content

Commit 3cff46d

Browse files
author
Marta
committed
Minor bugfix metrics calculation.
1 parent e022e0c commit 3cff46d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_wrapper/extra/evaluation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def multiclass_metrics(pred_list, verbose, extra_vars, split):
247247
# Compute accuracy
248248
top_n_accuracies = [3, 5]
249249
accuracy = sklearn_metrics.accuracy_score(y_gt, y_pred)
250-
acc_top_n = []
250+
acc_top_n = {}
251251
for topn in top_n_accuracies:
252252
acc_top_n[topn] = __top_k_accuracy(y_gt, y_pred, topn)
253253
# accuracy_balanced = sklearn_metrics.accuracy_score(y_gt, y_pred, sample_weight=sample_weights, )

0 commit comments

Comments
 (0)