Add plot titles and newstyle plt import in OLS vs Ridge example

ogrisel · ogrisel · commit 7430eba323fb · 2013-10-24T16:25:40.000+02:00
diff --git a/examples/linear_model/plot_ols_ridge_variance.py b/examples/linear_model/plot_ols_ridge_variance.py
@@ -29,7 +29,7 @@
 
 
 import numpy as np
-import pylab as pl
+import matplotlib.pyplot as plt
 
 from sklearn import linear_model
 
@@ -44,9 +44,10 @@
 
 fignum = 1
 for name, clf in classifiers.iteritems():
-    fig = pl.figure(fignum, figsize=(4, 3))
-    pl.clf()
-    ax = pl.axes([.12, .12, .8, .8])
+    fig = plt.figure(fignum, figsize=(4, 3))
+    plt.clf()
+    plt.title(name)
+    ax = plt.axes([.12, .12, .8, .8])
 
     for _ in range(6):
         this_X = .1 * np.random.normal(size=(2, 1)) + X_train
@@ -67,4 +68,4 @@
     ax.set_xlim(0, 2)
     fignum += 1
 
-pl.show()
+plt.show()
