FIX KernelRidge checks if fitted before predict

scikit-learn · Jan 18, 2015 · 2a2e7c6 · 2a2e7c6
1 parent 7223efd
commit 2a2e7c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sklearn/kernel_ridge.py b/sklearn/kernel_ridge.py
@@ -13,6 +13,7 @@
 from sklearn.base import BaseEstimator, RegressorMixin
 from sklearn.metrics.pairwise import pairwise_kernels
 from sklearn.linear_model.ridge import _solve_cholesky_kernel
+from sklearn.utils.validation import NotFittedError, check_is_fitted
 
 
 class KernelRidge(BaseEstimator, RegressorMixin):
@@ -178,5 +179,6 @@ def predict(self, X):
         C : array, shape = [n_samples] or [n_samples, n_targets]
             Returns predicted values.
         """
+        check_is_fitted(self, ["X_fit_", "dual_coef_"])
         K = self._get_kernel(X, self.X_fit_)
         return np.dot(K, self.dual_coef_)