From f0f9668e9de9e0c8c406384af1a76cf9931234e0 Mon Sep 17 00:00:00 2001 From: MechCoder <manojkumarsivaraj334@gmail.com> Date: Fri, 29 Aug 2014 14:26:15 +0200 Subject: [PATCH] FIX: Fix the imputation example --- examples/imputation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/imputation.py b/examples/imputation.py index aa04521de2..4ccdd41289 100644 --- a/examples/imputation.py +++ b/examples/imputation.py @@ -48,7 +48,7 @@ missing_features = rng.randint(0, n_features, n_missing_samples) # Estimate the score without the lines containing missing values X_filtered = X_full[~missing_samples, :] -y_filtered = y_full[~missing_samples, :] +y_filtered = y_full[~missing_samples] estimator = RandomForestRegressor(random_state=0, n_estimators=100) score = cross_val_score(estimator, X_filtered, y_filtered).mean() print("Score without the samples containing missing values = %.2f" % score) -- GitLab