Merge pull request scikit-learn#2935 from kevinpCroat/patch-1

jnothman · jnothman · commit 0a511827bea7 · 2014-03-05T10:38:19.000+11:00
FIX error in imputation example
diff --git a/examples/imputation.py b/examples/imputation.py
@@ -12,8 +12,8 @@
 Script output:
 
   Score with the entire dataset = 0.56
-  Score without the samples containing missing values = 0.48
-  Score after imputation of the missing values = 0.55
+  Score without the samples containing missing values = 0.49
+  Score after imputation of the missing values = 0.57
 
 """
 import numpy as np
@@ -48,7 +48,7 @@
 
 # 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)
