81e73bcd883ef83af9be3f5722d8bcc6ffa1146f,qanta/guesser/tfidf.py,TfidfGuesser,guess,#TfidfGuesser#,37

Before Change

        guess_scores = guess_matrix.max(axis=1)
        guess_indices = guess_matrix.argmax(axis=1)
        guesses = []
        for idx, s in zip(guess_indices, guess_scores):
            guesses.append([(self.i_to_ans[idx], s)])

        return guesses

    def save(self, directory: str) -> None:
        with open(os.path.join(directory, "params.pickle"), "wb") as f:

After Change

        guess_scores = guess_matrix.max(axis=1)
        guess_indices = guess_matrix.argmax(axis=1)
        guesses = []
        for i in range(len(questions)):
            idx = guess_indices[i]
            score = guess_indices[i]
            guesses.append([(self.i_to_ans[idx], score)])

        return guesses

    def save(self, directory: str) -> None:
        with open(os.path.join(directory, "params.pickle"), "wb") as f:

Instances