text_html_output_1.png
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
tsne_data = np.vstack((tsne_model.T, y)).T
tsne_df = pd.DataFrame(data=tsne_data, columns=('Dimension 1', 'Dimension 2', 'Class'))
sns.FacetGrid(tsne_df, height=8, hue='Class').map(plt.scatter, 'Dimension 1', 'Dimension 2').add_legend()