# Import TSNE
from sklearn.manifold import TSNE
from sklearn import datasets
import matplotlib.pyplot as plt

iris = datasets.load_iris()
iris_data = iris.data

species = iris.target

# Create a TSNE instance: model
model = TSNE(learning_rate=200)

# Apply fit_transform to samples: iris_data
tsne_features = model.fit_transform(iris_data)

# Select the 0th feature: xs
xs = tsne_features[:,0]

# Select the 1st feature: ys
ys = tsne_features[:,1]

# Scatter plot, coloring by variety_numbers
plt.scatter(xs, ys, c=species)
plt.show()