dimensionality_reduction_tsne.md

Note: This is a generated markdown export from the Jupyter notebook file dimensionality_reduction_tsne.ipynb. You can also view the notebook with the nbviewer from Jupyter.

Dimensionality Reduction with t-SNE

%matplotlib inline
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

from sklearn import manifold, datasets
from matplotlib.colors import ListedColormap

iris = datasets.load_iris()
tsne = manifold.TSNE(n_components=2, learning_rate=100)
new_dim = tsne.fit_transform(iris.data)

df = pd.DataFrame(new_dim, columns=['X', 'Y'])
df['label'] = iris.target
df.head()

	X	Y	label
0	-11.520270	-19.948196	0
1	-10.830556	-22.377188	0
2	-11.455172	-22.781273	0
3	-11.699793	-23.005066	0
4	-11.125145	-19.888693	0

fig = plt.figure()
fig.suptitle('t-SNE', fontsize=14, fontweight='bold')
ax = fig.add_subplot(111)

plt.scatter(df[df.label == 0].X, df[df.label == 0].Y, color='red', label=iris.target_names[0])
plt.scatter(df[df.label == 1].X, df[df.label == 1].Y, color='blue', label=iris.target_names[1])
plt.scatter(df[df.label == 2].X, df[df.label == 2].Y, color='green', label=iris.target_names[2])

_ = plt.legend(bbox_to_anchor=(1.25, 1))