EXA Fix issue in Statistical comparison of models using grid search (s…

…cikit-learn#19966)
j3nnn1 · Jun 21, 2021 · c3a3b1e · c3a3b1e
1 parent bb6117b
commit c3a3b1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/model_selection/plot_grid_search_stats.py b/examples/model_selection/plot_grid_search_stats.py
@@ -190,9 +190,9 @@ def corrected_std(differences, n_train, n_test):
     corrected_std : int
         Variance-corrected standard deviation of the set of differences.
     """
-    n = n_train + n_test
+    kr = len(differences)
     corrected_var = (
-        np.var(differences, ddof=1) * ((1 / n) + (n_test / n_train))
+        np.var(differences, ddof=1) * (1 / kr + n_test / n_train)
     )
     corrected_std = np.sqrt(corrected_var)
     return corrected_std