def METHOD_NAME(self, X, delayed=True):
    """
    Make predictions for X and returns a dask collection.
    Parameters
    ----------
    X : Dask cuDF dataframe  or CuPy backed Dask Array (n_rows, n_features)
        Distributed dense matrix (floats or doubles) of shape
        (n_samples, n_features).
    delayed : bool (default = True)
        Whether to do a lazy prediction (and return Delayed objects) or an
        eagerly executed one.
    Returns
    -------
    y : Dask cuDF dataframe  or CuPy backed Dask Array (n_rows, 1)
    """
    return self._predict(X, delayed=delayed)