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

* remove beta from F1

* remove from functional

Co-authored-by: Teddy Koker <teddy.koker@gmail.com>
This commit is contained in:
Roger Shieh 2020-12-12 15:14:17 +08:00 committed by GitHub
parent 0de43d1cb6
commit b5a2afd232
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 2 deletions

View File

@ -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,

View File

@ -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