'Compute predictions of y from X.
Parameters
X : array, shape (n_samples, n_features)
The data used to compute the predictions.
Returns
y_pred : array, shape (n_samples,)
The predictions.'

def predict(self, X):

    return self.model.predict(X)
