def update_results_from_preds()

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