matplotlib new figure creation modified

aarshayj · thismlguy · commit f4d8a64abbf1 · 2018-01-12T17:45:11.000+05:30
diff --git a/sklearn/plot/_gridsearch_results.py b/sklearn/plot/_gridsearch_results.py
@@ -6,7 +6,8 @@ def _plot_1D_results(cv_results, params, ax, xlabel, ylabel, title,
                      fmt, xtickrotation):
     import matplotlib.pyplot as plt
     if ax is None:
-        ax = plt.gca()
+        fig = plt.figure()
+        ax = fig.add_subplot(111)
 
     param = params[0]
     param_range = sorted(cv_results['param_%s' % param])
@@ -40,6 +41,7 @@ def _plot_1D_results(cv_results, params, ax, xlabel, ylabel, title,
     ylabel = "Score" if ylabel is None else ylabel
     ax.set_ylabel(ylabel)
     ax.set_title(title)
+    plt.draw()
     return img
 
 
diff --git a/sklearn/plot/_heatmap.py b/sklearn/plot/_heatmap.py
@@ -52,7 +52,8 @@ def plot_heatmap(values, xlabel="", ylabel="", xticklabels=None,
 
     import matplotlib.pyplot as plt
     if ax is None:
-        ax = plt.gca()
+        fig = plt.figure()
+        ax = fig.add_subplot(111)
     img = ax.pcolormesh(values, cmap=cmap, vmin=vmin, vmax=vmax, norm=norm)
 
     # this will allow us to access the pixel values:
@@ -98,4 +99,6 @@ def plot_heatmap(values, xlabel="", ylabel="", xticklabels=None,
     if title is not None:
         ax.set_title(title)
 
+    plt.draw()
+
     return img
