We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2be6ccf commit 2accbe4Copy full SHA for 2accbe4
tests/test_metrics.py
@@ -59,8 +59,11 @@ def test_recall():
59
recall_micro = Recall(7)
60
recall_macro = Recall(7, macro_averaging=True)
61
62
- recall_micro_score = recall_micro(y_true, logits)
63
- recall_macro_score = recall_macro(y_true, logits)
+ recall_micro(y_true, logits)
+ recall_macro(y_true, logits)
64
+
65
+ recall_micro_score = recall_micro.__returnmetric__()
66
+ recall_macro_score = recall_macro.__returnmetric__()
67
68
assert isinstance(recall_micro_score, torch.Tensor), "Expected a tensor output."
69
assert isinstance(recall_macro_score, torch.Tensor), "Expected a tensor output."
0 commit comments