aboutsummaryrefslogtreecommitdiff
path: root/gensvm/sklearn_util.py
diff options
context:
space:
mode:
authorGertjan van den Burg <gertjanvandenburg@gmail.com>2019-12-04 13:08:42 +0000
committerGertjan van den Burg <gertjanvandenburg@gmail.com>2019-12-04 13:09:02 +0000
commit0361b805330cbf72263af025a2267b196456f715 (patch)
treea042049071702163d30bce82f9523d85714b1bac /gensvm/sklearn_util.py
parentRemove is_multimetric flag (diff)
downloadpygensvm-0361b805330cbf72263af025a2267b196456f715.tar.gz
pygensvm-0361b805330cbf72263af025a2267b196456f715.zip
Deal with changing behavior of check_is_fitted
This is going to change in version 0.23, so we might as well inline it now.
Diffstat (limited to 'gensvm/sklearn_util.py')
-rw-r--r--gensvm/sklearn_util.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/gensvm/sklearn_util.py b/gensvm/sklearn_util.py
index 182f257..e23921b 100644
--- a/gensvm/sklearn_util.py
+++ b/gensvm/sklearn_util.py
@@ -207,7 +207,12 @@ def _skl_check_is_fitted(estimator, method_name, refit):
"attribute" % (type(estimator).__name__, method_name)
)
else:
- check_is_fitted(estimator, "best_estimator_")
+ if not hasattr(estimator, "best_estimator_"):
+ raise NotFittedError(
+ "This %s instance is not fitted yet. Call "
+ "'fit' with appropriate arguments before using this "
+ "estimator." % type(estimator).__name__
+ )
def _skl_grid_score(X, y, scorer_, best_estimator_, refit, multimetric_):