DatasetReport shift_dict bug fixed

This commit is contained in:
Lorenzo Volpi 2023-10-28 16:14:55 +02:00
parent 345807977c
commit 3d04aa660d
1 changed files with 8 additions and 0 deletions

View File

@ -280,6 +280,9 @@ class DatasetReport:
return np.around([(1.0 - p, p) for p in self.prevs], decimals=2)
def add(self, cr: CompReport):
if cr is None:
return
self.crs.append(cr)
if self._dict is None:
@ -320,6 +323,11 @@ class DatasetReport:
[self.s_dict[col][sp], cr_s_dict[col][sp]]
)
for sp in self.s_prevs:
for col, vals in self.s_dict.items():
if sp not in vals:
vals[sp] = []
for k, score in cr.fit_scores.items():
if k not in self.fit_scores:
self.fit_scores[k] = []