Skip to content

Commit

Permalink
Update regex syntax for python3.12
Browse files Browse the repository at this point in the history
  • Loading branch information
Eleven1Liu committed May 25, 2024
1 parent bb4481f commit 7ebfe0e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions libmultilabel/linear/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -300,13 +300,13 @@ def get_metrics(monitor_metrics: list[str], num_classes: int, multiclass: bool =
monitor_metrics = []
metrics = {}
for metric in monitor_metrics:
if re.match("P@\d+", metric):
if re.match(r"P@\d+", metric):
metrics[metric] = PrecisionAtK(top_k=int(metric[2:]))
elif re.match("R@\d+", metric):
elif re.match(r"R@\d+", metric):
metrics[metric] = RecallAtK(top_k=int(metric[2:]))
elif re.match("RP@\d+", metric):
elif re.match(r"RP@\d+", metric):
metrics[metric] = RPrecisionAtK(top_k=int(metric[3:]))
elif re.match("NDCG@\d+", metric):
elif re.match(r"NDCG@\d+", metric):
metrics[metric] = NDCGAtK(top_k=int(metric[5:]))
elif metric in {"Another-Macro-F1", "Macro-F1", "Micro-F1"}:
metrics[metric] = F1(num_classes, average=metric[:-3].lower(), multiclass=multiclass)
Expand Down

0 comments on commit 7ebfe0e

Please sign in to comment.