added import np statements

jwkvam · Mar 31, 2015 · 9e7a660 · 9e7a660
1 parent 22963a0
commit 9e7a660
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/doc/modules/linear_model.rst b/doc/modules/linear_model.rst
@@ -1073,6 +1073,7 @@ polynomial regression can be created and used as follows::
     >>> from sklearn.preprocessing import PolynomialFeatures
     >>> from sklearn.linear_model import LinearRegression
     >>> from sklearn.pipeline import Pipeline
+    >>> import numpy as np
     >>> model = Pipeline([('poly', PolynomialFeatures(degree=3)),
     ...                   ('linear', LinearRegression(fit_intercept=False))])
     >>> # fit to an order-3 polynomial data
@@ -1098,6 +1099,7 @@ This way, we can solve the XOR problem with a linear classifier::
 
     >>> from sklearn.linear_model import Perceptron
     >>> from sklearn.preprocessing import PolynomialFeatures
+    >>> import numpy as np
     >>> X = np.array([[0, 0], [0, 1], [1, 0], [1, 1]])
     >>> y = X[:, 0] ^ X[:, 1]
     >>> X = PolynomialFeatures(interaction_only=True).fit_transform(X)