This notebook contains an excerpt from the book Machine Learning for OpenCV by Michael Beyeler. The code is released under the MIT license, and is available on GitHub.

Note that this excerpt contains only the raw code - the book is rich with additional explanations and illustrations. If you find this content useful, please consider supporting the work by buying the book!

< Dealing with Data Using Python's NumPy Package | Contents | Visualizing Data Using Matplotlib >

Loading External Datasets in Python

Thanks to the SciPy community, there are tons of resources out there for getting your hands on some data.

A particularly useful resource comes in the form of the sklearn.datasets package of scikit-learn. This package comes pre-installed with some small datasets that do not require to download any files from external websites. These datasets include:

load_boston: The Boston dataset contains housing prices in different suburbs of Boston, along with a number of interesting features such as per capita crime rate by town, proportion of residential land and non-retail business, etc.
load_iris: The Iris dataset contains three different types of iris flowers (setosa, versicolor, and virginica), along with four features describing the width and length of the sepals and petals.
load_diabetes: The diabetes dataset lets you classify patients as having diabetes or not, based on features such as patient age, sex, body mass index, average blood pressure, and six blood serum measurements.
load_digits: The digits dataset contains 8x8 pixel images of digits 0-9.
load_linnerud: The Linnerud dataset contains three physiological and three exercies variables measured on twenty middle-aged men in a fitness club.

Even better, scikit-learn allows you to download datasets directly from external repositories, such as:

fetch_olivetti_faces: The Olivetta face dataset contains of ten different images each of 40 distinct subjects.
fetch_20newsgroups: The 20 newsgroup dataset contains around 18,000 newsgroups post on 20 topics.

Even better, it is possible to download datasets directly from the machine learning database at http://mldata.org.

For example, to download the MNIST dataset of handwritten digits, simply type:



In [1]:

    
from sklearn import datasets



In [2]:

    
mnist = datasets.fetch_mldata('MNIST original')

Note that this might take a while, depending on your internet connection.

The MNIST database contains a total of 70,000 examples of handwritten digits (28x28 pixel images, labeled from 0 to 9). Data and labels are delivered in two separate containers, which we can inspect as follows:



In [3]:

    
mnist.data.shape









    Out[3]:





(70000, 784)



In [4]:

    
mnist.target.shape









    Out[4]:





(70000,)

Here, we can see that mnist.data contains 70,000 images of 28 x 28 = 784 pixels each. Labels are stored in mnist.target, where there is only one label per image.

We can further inspect the values of all targets, but we don't just want to print them all. Instead, we are interested to see all distinct target values, which is easy to do with NumPy:



In [5]:

    
import numpy as np



In [6]:

    
np.unique(mnist.target)









    Out[6]:





array([ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9.])

< Dealing with Data Using Python's NumPy Package | Contents | Visualizing Data Using Matplotlib >