We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a4df0f2 commit 2a85e81Copy full SHA for 2a85e81
utils/metrics/precision.py
@@ -17,7 +17,7 @@ def __init__(self, num_classes: int, micro_averaging: bool = False):
17
super().__init__()
18
19
self.num_classes = num_classes
20
- self._micro_averaging = micro_averaging
+ self.micro_averaging = micro_averaging
21
22
def forward(self, y_true: torch.tensor, y_pred: torch.tensor) -> torch.tensor:
23
"""Compute precision of model
0 commit comments