Skip to content

Commit 2a85e81

Browse files
committed
fixed bug in test-file
1 parent a4df0f2 commit 2a85e81

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/metrics/precision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def __init__(self, num_classes: int, micro_averaging: bool = False):
1717
super().__init__()
1818

1919
self.num_classes = num_classes
20-
self._micro_averaging = micro_averaging
20+
self.micro_averaging = micro_averaging
2121

2222
def forward(self, y_true: torch.tensor, y_pred: torch.tensor) -> torch.tensor:
2323
"""Compute precision of model

0 commit comments

Comments
 (0)