in expanded_checklist/checklist/eval_core.py [0:0]
def update_results_from_preds(self, preds, confs):
"""Updates results from preds and confs
Assumes that example_lists_and_indices or to_raw_examples or
to_raw_file was called before, so that self.result_indexes exists
Parameters
----------
preds : list
Predictions
confs : list
Confidences
Updates self.results.preds and self.results.confs
"""
result_indexes = self.result_indexes
if is_1d_list(self.data[0]):
self.results.preds = [[] for _ in self.data]
self.results.confs = [[] for _ in self.data]
for i, p, c in zip(result_indexes, preds, confs):
self.results.preds[i].append(p)
self.results.confs[i].append(c)
elif is_2d_list(self.data[0]):
self.results.preds = [[[] for _ in x] for x in self.data]
self.results.confs = [[[] for _ in x] for x in self.data]
for (i, j), p, c in zip(result_indexes, preds, confs):
self.results.preds[i][j].append(p)
self.results.confs[i][j].append(c)
else:
self.results.preds = [None for _ in self.data]
self.results.confs = [None for _ in self.data]
for i, p, c in zip(result_indexes, preds, confs):
self.results.preds[i] = p
self.results.confs[i] = c