In [ ]:
%matplotlib inline

Visualizing the digits dataset using UMAP

This example loads in some data from the scikit-learn digits dataset and plots it using UMAP.


In [ ]:
# Code source: Andrew Heusser and Leland McInnes
# License: MIT

from sklearn import datasets
import hypertools as hyp

digits = datasets.load_digits(n_class=6)
data = digits.data
hue = digits.target.astype('str')

hyp.plot(data, '.', reduce='UMAP', hue=hue, ndims=2)