811554e63022cc746fd1b0b7aa2ef5ef7a02f376,dask_ml/wrappers.py,ParallelPostFit,predict,#ParallelPostFit#Any#,255

Before Change

        -------
        y : array-like
        
        predict = self._check_method("predict")
        X = self._check_array(X)

        if isinstance(X, da.Array):
            return X.map_blocks(predict, dtype="int", drop_axis=1)

        elif isinstance(X, dd._Frame):
            return _apply_partitionwise(X, predict)

        else:
            return predict(X)

After Change

        -------
        y : array-like
        
        self._check_method("predict")
        X = self._check_array(X)

        if isinstance(X, da.Array):
            result = X.map_blocks(
                _predict, dtype="int", estimator=self._postfit_estimator, drop_axis=1
            )
            return result

        elif isinstance(X, dd._Frame):
            return X.map_partitions(
                _predict, estimator=self._postfit_estimator, meta=np.array([1])
            )

        else:
            return _predict(X, estimator=self._postfit_estimator)

    def predict_proba(self, X):
        Predict for X.

Instances