Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
williamFalcon committed Jun 16, 2020
1 parent 6204316 commit 2d9b90b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pytorch_lightning/metrics/functional/classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -626,6 +626,7 @@ def precision_recall_curve(
>>> precision, recall, thresholds = precision_recall_curve(pred, target)
>>> precision_recall_curve(pred, target)
(tensor([0.3333, 0.0000, 0.0000, 1.0000]), tensor([1., 0., 0., 0.]), tensor([1., 2., 3.]))
"""
fps, tps, thresholds = _binary_clf_curve(pred=pred, target=target,
sample_weight=sample_weight,
Expand Down

0 comments on commit 2d9b90b

Please sign in to comment.