e604607e0a26bd5ca244b60dba8769779f2f07a4,qanta/guesser/dan.py,DanGuesser,_guess_batch,#DanGuesser#Any#Any#,462

Before Change

        out = self.model(input_dict, lengths_dict, qnums)
        probs = F.softmax(out)
        scores, preds = torch.max(probs, 1)
        scores = scores.data.cpu().numpy()
        preds = preds.data.cpu().numpy()

        for p, s in zip(preds, scores):
            guesses.append([(self.i_to_ans[p], s)])

After 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([])
            for p in ps:

Instances