cb89711536f566f3ed7d121520fe32ad41ca83bc,finetune/association.py,Association,predict,#Association#Any#,136

Before Change

            print("probs shape is ", pred["association_probs"].shape)

            most_likely_associations = np.argmax(pred["association"], 0)
            most_likely_class_id = pred["association"][range(len(most_likely_associations)), tuple(most_likely_associations)]
            print([prob[idx, cls] for prob, idx, cls in zip(pred["association_probs"], most_likely_associations, most_likely_class_id)])

            associations.append((
                most_likely_associations,

After Change

            labels.append(pred_labels)
            batch_probas.append(pred["sequence_probs"])
            pred["association_probs"] = self.prune_probs(pred["association_probs"], pred_labels)
            most_likely_associations, most_likely_class_id = zip(*[np.unravel_index(np.argmax(a, axis=None), a.shape) for a in pred["association_probs"]])
            associations.append((
                most_likely_associations,
                self.input_pipeline.association_encoder.inverse_transform(most_likely_class_id),
                [prob[idx, cls] for prob, idx, cls in zip(pred["association_probs"], most_likely_associations, most_likely_class_id)]

Instances