Skip to content

Commit 1b171b7

Browse files
authored
Fixing __reset__ method for precision
When used in the precision did not reset itself correctly, resulting in precision = 1 forall epochs > 1. @Johanmkr
1 parent 546bead commit 1b171b7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

CollaborativeCoding/metrics/precision.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,8 @@ def __reset__(self):
121121
None
122122
Returns None
123123
"""
124-
self.y_true = self.y_pred = []
124+
self.y_true = []
125+
self.y_pred = []
125126
return None
126127

127128

0 commit comments

Comments
 (0)