e79449afe45f4da6f1756a7d92bdd3cd6618c638,qanta/guesser/dan.py,DanGuesser,_guess_batch,#DanGuesser#,434

Before Change

        out = self.model(input_dict, lengths_dict, qnums)
        probs = [p for p in F.softmax(out).data.cpu().numpy()]
        logits = [l for l in out.data.cpu().numpy()]
        preds = [np.argsort(-l)[:10] for l in logits]
        guesses = []
        for i, ps in enumerate(preds):
            guesses.append([])

After Change

        out = self.model(input_dict, lengths_dict, qnums)
        probs = F.softmax(out).data.cpu().numpy()
        n_examples = probs.shape[0]
        preds = np.argsort(-probs, axis=1)
        for i in range(n_examples):
            guesses.append([])
            for p in preds[i][:max_n_guesses]:
                guesses[-1].append((self.i_to_ans[p], probs[i][p]))

Instances