From b5a2afd232bf6dea2e74200038501d858c59a49f Mon Sep 17 00:00:00 2001 From: Roger Shieh Date: Sat, 12 Dec 2020 15:14:17 +0800 Subject: [PATCH] Remove beta arg from F1 class and functional (#5076) * remove beta from F1 * remove from functional Co-authored-by: Teddy Koker --- pytorch_lightning/metrics/classification/f_beta.py | 1 - pytorch_lightning/metrics/functional/f_beta.py | 1 - 2 files changed, 2 deletions(-) diff --git a/pytorch_lightning/metrics/classification/f_beta.py b/pytorch_lightning/metrics/classification/f_beta.py index 85a0d3f736bac..fadfd000ebbe1 100755 --- a/pytorch_lightning/metrics/classification/f_beta.py +++ b/pytorch_lightning/metrics/classification/f_beta.py @@ -212,7 +212,6 @@ class F1(FBeta): def __init__( self, num_classes: int = 1, - beta: float = 1.0, threshold: float = 0.5, average: str = "micro", multilabel: bool = False, diff --git a/pytorch_lightning/metrics/functional/f_beta.py b/pytorch_lightning/metrics/functional/f_beta.py index 5f6962d2b3964..2b0ba194d56f0 100755 --- a/pytorch_lightning/metrics/functional/f_beta.py +++ b/pytorch_lightning/metrics/functional/f_beta.py @@ -110,7 +110,6 @@ def f1( preds: torch.Tensor, target: torch.Tensor, num_classes: int, - beta: float = 1.0, threshold: float = 0.5, average: str = "micro", multilabel: bool = False