Skip to content

Commit

Permalink
docs
Browse files Browse the repository at this point in the history
  • Loading branch information
elephantmipt committed Nov 3, 2020
1 parent b32b355 commit c2c9fee
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions catalyst/metrics/tests/test_fbeta_precision_recall.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,21 +51,21 @@ def test_precision_recall_fbeta_support_binary(
Test for precision_recall_fbeta_support.
Args:
outputs:
targets:
precision_true:
recall_true:
fbeta_true:
support_true:
outputs: test arg
targets: test arg
precision_true: test arg
recall_true: test arg
fbeta_true: test arg
support_true: test arg
"""
(
precision_score,
recall_score,
fbeta_score_,
fbeta_score_ev,
support,
) = precision_recall_fbeta_support(outputs=outputs, targets=targets)

assert torch.isclose(precision_score[1], torch.tensor(precision_true))
assert torch.isclose(recall_score[1], torch.tensor(recall_true))
assert torch.isclose(fbeta_score_[1], torch.tensor(fbeta_true))
assert torch.isclose(fbeta_score_ev[1], torch.tensor(fbeta_true))
assert support[1] == support_true

0 comments on commit c2c9fee

Please sign in to comment.