Fix bug in metric wrapper

c-salomonsen · c-salomonsen · commit b8c076f8eb54 · 2025-02-20T13:07:50.000+01:00
diff --git a/CollaborativeCoding/load_metric.py b/CollaborativeCoding/load_metric.py
@@ -107,7 +107,7 @@ def resetmetric(self):
         y_pred = th.rand((5, class_size))
 
         metricwrapper = MetricWrapper(
-            metric,
+            *metrics,
             num_classes=class_size,
             macro_averaging=True if class_size % 2 == 0 else False,
         )

Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ def resetmetric(self):`
`107`	`107`	`y_pred = th.rand((5, class_size))`
`108`	`108`
`109`	`109`	`metricwrapper = MetricWrapper(`
`110`		`- metric,`
	`110`	`+ *metrics,`
`111`	`111`	`num_classes=class_size,`
`112`	`112`	`macro_averaging=True if class_size % 2 == 0 else False,`
`113`	`113`	`)`