'Predict using the estimated model.
This is a wrapper for `estimator_.predict(X)`.
Parameters
X : numpy array of shape [n_samples, n_features]
Returns
y : array, shape = [n_samples] or [n_samples, n_targets]
Returns predicted values.'

def predict(self, X):

    check_is_fitted(self, 'estimator_')
    return self.estimator_.predict(X)
