From 26ebd8dc417f074adead192daa409eee9cf187ff Mon Sep 17 00:00:00 2001 From: Roger Shieh Date: Fri, 11 Dec 2020 13:13:37 +0800 Subject: [PATCH 1/2] remove beta from F1 --- pytorch_lightning/metrics/classification/f_beta.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_lightning/metrics/classification/f_beta.py b/pytorch_lightning/metrics/classification/f_beta.py index 56cc00f9a5dce..1dcac3a898a27 100755 --- a/pytorch_lightning/metrics/classification/f_beta.py +++ b/pytorch_lightning/metrics/classification/f_beta.py @@ -183,7 +183,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, From 8d612e016e099dcacefb4a332ad284fa4ee34977 Mon Sep 17 00:00:00 2001 From: Roger Shieh Date: Fri, 11 Dec 2020 13:38:05 +0800 Subject: [PATCH 2/2] remove from functional --- pytorch_lightning/metrics/functional/f_beta.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_lightning/metrics/functional/f_beta.py b/pytorch_lightning/metrics/functional/f_beta.py index 3f0a7a0449325..f4e1331888b62 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