892065c8339f9c20d36127645ba3ae169577f47b,skorch/classifier.py,NeuralNetClassifier,predict,#NeuralNetClassifier#,179

Before Change

        y_preds = []
        for yp in self.forward_iter(X, training=False):
            yp = yp[0] if isinstance(yp, tuple) else yp
            y_preds.append(to_numpy(yp.max(-1)[-1]))
        y_pred = np.concatenate(y_preds, 0)
        return y_pred

After Change

        y_pred : numpy ndarray

        
        return super().predict_proba(X).argmax(1)


neural_net_binary_clf_doc_start = NeuralNet for binary classification tasks

Instances