%pylab inline
Populating the interactive namespace from numpy and matplotlib
from sklearn import datasets
from sklearn.linear_model import LogisticRegression
from sklearn.preprocessing import PolynomialFeatures
from sklearn.cross_validation import cross_val_score
iris = datasets.load_iris()
linear_classifier = LogisticRegression()
linear_scores = cross_val_score(linear_classifier, \
iris.data, iris.target, cv=10) #2
print "Accuracy (linear):\t%0.2f (+/- %0.2f)" % \
(linear_scores.mean(), linear_scores.std() * 2)
pol = PolynomialFeatures(degree=2)
nonlinear_data = pol.fit_transform(iris.data)
nonlinear_classifier = LogisticRegression()
nonlinear_scores = cross_val_score(nonlinear_classifier, \
nonlinear_data, iris.target, cv=10)
print "Accuracy (nonlinear):\t%0.2f (+/- %0.2f)" % \
(nonlinear_scores.mean(), nonlinear_scores.std() * 2)
Accuracy (linear): 0.95 (+/- 0.12) Accuracy (nonlinear): 0.98 (+/- 0.09)