fix examples

Author: Dominik Waurenschk

examples/plot_mpg.py

@@ -51,7 +51,7 @@
 plt.show()
 
 # Calculate the variance
-mpg_V_IJ_unbiased = fci.random_forest_error(mpg_forest, mpg_X_train,
+mpg_V_IJ_unbiased = fci.random_forest_error(mpg_forest, mpg_X_train.shape,
                                             mpg_X_test)
 
 # Plot error bars for predicted MPG using unbiased variance

examples/plot_mpg_svr.py

@@ -52,7 +52,7 @@
 plt.show()
 
 # Calculate the variance
-mpg_V_IJ_unbiased = fci.random_forest_error(mpg_bagger, mpg_X_train, mpg_X_test)
+mpg_V_IJ_unbiased = fci.random_forest_error(mpg_bagger, mpg_X_train.shape, mpg_X_test)
 
 # Plot error bars for predicted MPG using unbiased variance
 plt.errorbar(mpg_y_test, mpg_y_hat, yerr=np.sqrt(mpg_V_IJ_unbiased), fmt="o")

examples/plot_spam.py

@@ -44,7 +44,7 @@
 plt.legend()
 
 # Calculate the variance
-spam_V_IJ_unbiased = fci.random_forest_error(spam_RFC, spam_X_train,
+spam_V_IJ_unbiased = fci.random_forest_error(spam_RFC, spam_X_train.shape,
                                              spam_X_test)
 
 # Plot forest prediction for emails and standard deviation for estimates