Skip to content

Commit 86be2c9

Browse files
committed
fixed invalid arguments
1 parent b4e0d4d commit 86be2c9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

libmultilabel/linear/metrics.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -297,9 +297,9 @@ def get_metrics(monitor_metrics: list[str], num_classes: int, multiclass: bool =
297297
metrics = {}
298298
for metric in monitor_metrics:
299299
if re.match("P@\d+", metric):
300-
metrics[metric] = PrecisionAtK(num_classes, average="samples", top_k=int(metric[2:]))
300+
metrics[metric] = PrecisionAtK(top_k=int(metric[2:]))
301301
elif re.match("R@\d+", metric):
302-
metrics[metric] = RecallAtK(num_classes, average="samples", top_k=int(metric[2:]))
302+
metrics[metric] = RecallAtK(top_k=int(metric[2:]))
303303
elif re.match("RP@\d+", metric):
304304
metrics[metric] = RPrecisionAtK(top_k=int(metric[3:]))
305305
elif re.match("NDCG@\d+", metric):

0 commit comments

Comments
 (0)