DatasetReport shift_dict bug fixed
This commit is contained in:
parent
345807977c
commit
3d04aa660d
|
@ -280,6 +280,9 @@ class DatasetReport:
|
||||||
return np.around([(1.0 - p, p) for p in self.prevs], decimals=2)
|
return np.around([(1.0 - p, p) for p in self.prevs], decimals=2)
|
||||||
|
|
||||||
def add(self, cr: CompReport):
|
def add(self, cr: CompReport):
|
||||||
|
if cr is None:
|
||||||
|
return
|
||||||
|
|
||||||
self.crs.append(cr)
|
self.crs.append(cr)
|
||||||
|
|
||||||
if self._dict is None:
|
if self._dict is None:
|
||||||
|
@ -320,6 +323,11 @@ class DatasetReport:
|
||||||
[self.s_dict[col][sp], cr_s_dict[col][sp]]
|
[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():
|
for k, score in cr.fit_scores.items():
|
||||||
if k not in self.fit_scores:
|
if k not in self.fit_scores:
|
||||||
self.fit_scores[k] = []
|
self.fit_scores[k] = []
|
||||||
|
|
Loading…
Reference in New Issue