Remove beta arg from F1 class and functional (Lightning-AI#5076)

* remove beta from F1 * remove from functional Co-authored-by: Teddy Koker <teddy.koker@gmail.com>
gianscarpe · Dec 12, 2020 · b5a2afd · b5a2afd
1 parent 0de43d1
commit b5a2afd
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git 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
@@ -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