import CovergenceWarning

intel · Dec 6, 2018 · bf1d7c8 · bf1d7c8
1 parent 98af1f7
commit bf1d7c8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/daal4py/sklearn/monkeypatch/logistic_path.py b/daal4py/sklearn/monkeypatch/logistic_path.py
@@ -31,6 +31,7 @@
                            check_random_state)
 from sklearn.linear_model.sag import sag_solver
 from sklearn.utils.optimize import newton_cg
+from sklearn.exceptions import ConvergenceWarning
 from sklearn.linear_model.logistic import (_check_solver,
                                            _check_multi_class,
                                            _fit_liblinear,