3e6bb1698d48305eb714f3b994317cd171e72b56,src/allennlp_mods/correlation.py,Correlation,__call__,#Correlation#Any#Any#,34

Before Change

        if isinstance(predictions, np.ndarray):
            predictions = predictions.tolist()
        if isinstance(labels, np.ndarray):
            labels = labels.tolist()

        if hasattr(self, "_per_batch_history"):
            batch_corr = self._correlation(labels, predictions)
            self._per_batch_history.append(batch_corr)

After Change

        if isinstance(predictions, torch.Tensor):
            predictions = predictions.cpu().numpy()
        if isinstance(labels, torch.Tensor):
            labels = labels.cpu().numpy()

        // Verify shape match
        assert predictions.shape == labels.shape, ("Predictions and labels must"
                                                   " have matching shape. Got:"
                                                   " preds=%s, labels=%s" % (
                                                       str(predictions.shape),
                                                       str(labels.shape)))
        predictions = list(predictions.flatten())
        labels = list(labels.flatten())

        if hasattr(self, "_per_batch_history"):
            batch_corr = self._correlation(labels, predictions)
            self._per_batch_history.append(batch_corr)

Instances