Update GANDLF/metrics/classification.py

szmazurek · sarthakpati · web-flow · commit c6da07a60234 · 2023-11-22T15:10:29.000+01:00
Co-authored-by: Sarthak Pati &lt;patis@iu.edu&gt;
diff --git a/GANDLF/metrics/classification.py b/GANDLF/metrics/classification.py
@@ -28,7 +28,7 @@ def overall_stats(predictions, ground_truth, params):
         "per_class_average": "macro",
         "per_class_weighted": "weighted",
     }
-    task = determine_task(params)
+    task = determine_classification_task_type(params)
     # consider adding a "multilabel field in the future"
     # metrics that need the "average" parameter
 

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def overall_stats(predictions, ground_truth, params):`
`28`	`28`	`"per_class_average": "macro",`
`29`	`29`	`"per_class_weighted": "weighted",`
`30`	`30`	`}`
`31`		`- task = determine_task(params)`
	`31`	`+ task = determine_classification_task_type(params)`
`32`	`32`	`# consider adding a "multilabel field in the future"`
`33`	`33`	`# metrics that need the "average" parameter`
`34`	`34`