Сначала я использую модель LDA, чтобы разделить мои данные на десять тем:
from sklearn.decomposition import LatentDirichletAllocation
lda = LatentDirichletAllocation(n_components=10,max_iter=10, n_jobs=-1,random_state=95865)
lda_trans = lda.fit_transform(X_tfidf)
Для визуализации темы я использую TSNE для преобразования данных в 2D,
topic_proportions = lda.transform(X_tfidf)[:1000]
from sklearn.manifold import TSNE
tsne_model = TSNE(n_components =2, learning_rate=800,angle=.99, init='pca')
topic_tsne_lda = tsne_model.fit_transform(topic_proportions)
plt.scatter(topic_tsne_lda[:,0],topic_tsne_lda[:,1])
И разбросанная картинка выглядит так:
Мой вопрос: как раскрасить кластер в разные цвета? Спасибо !!