Show Sidebar Hide Sidebar

Digits Classification Exercise in Scikit-learn

A tutorial exercise regarding the use of classification techniques on the Digits dataset.

This exercise is used in the Classification part of the Supervised learning: predicting an output variable from high-dimensional observations section of the A tutorial on statistical-learning for scientific data processing.

New to Plotly?

Plotly's Python library is free and open source! Get started by downloading the client and reading the primer.
You can set up Plotly to work in online or offline mode, or in jupyter notebooks.
We also have a quick-reference cheatsheet (new!) to help you get started!


In [1]:
import sklearn


In [2]:

from sklearn import datasets, neighbors, linear_model
Automatically created module for IPython interactive environment


In [4]:
digits = datasets.load_digits()
X_digits =
y_digits =

n_samples = len(X_digits)

X_train = X_digits[:.9 * n_samples]
y_train = y_digits[:.9 * n_samples]
X_test = X_digits[.9 * n_samples:]
y_test = y_digits[.9 * n_samples:]

knn = neighbors.KNeighborsClassifier()
logistic = linear_model.LogisticRegression()

print('KNN score: %f' %, y_train).score(X_test, y_test))
print('LogisticRegression score: %f'
      %, y_train).score(X_test, y_test))
KNN score: 0.961111
LogisticRegression score: 0.938889
Still need help?
Contact Us

For guaranteed 24 hour response turnarounds, upgrade to a Developer Support Plan.