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