From 4fccc53a5d05b0aa31d615754267405c07b6211a Mon Sep 17 00:00:00 2001 From: Dillon Niederhut Date: Sat, 13 Jul 2019 11:36:21 -0500 Subject: [PATCH] DOC: fix convergence warning in iterative example The estimator fails to converge reliably even with 1e4 iterations. We can keep the default max_iter and avoid convergence warnings increasing the tolerance. --- examples/impute/plot_iterative_imputer_variants_comparison.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/impute/plot_iterative_imputer_variants_comparison.py b/examples/impute/plot_iterative_imputer_variants_comparison.py index 06fab08c381f2..460b1c802cae1 100644 --- a/examples/impute/plot_iterative_imputer_variants_comparison.py +++ b/examples/impute/plot_iterative_imputer_variants_comparison.py @@ -105,7 +105,7 @@ score_iterative_imputer = pd.DataFrame() for impute_estimator in estimators: estimator = make_pipeline( - IterativeImputer(random_state=0, estimator=impute_estimator), + IterativeImputer(random_state=0, estimator=impute_estimator, tol=1), br_estimator ) score_iterative_imputer[impute_estimator.__class__.__name__] = \