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:
parent
0de43d1cb6
commit
b5a2afd232
|
@ -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,
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue