Я попробовал код по этой ссылке построить документ tfidf 2D graph
from sklearn.feature_extraction.text import CountVectorizer,
TfidfTransformer
from sklearn.decomposition import PCA
from sklearn.pipeline import Pipeline
import matplotlib.pyplot as plt
pipeline = Pipeline([
('vect', CountVectorizer()),
('tfidf', TfidfTransformer()),
])
X = pipeline.fit_transform(x_test).todense()
pca = PCA(n_components=2).fit(X)
data2D = pca.transform(X)
plt.scatter(data2D[:,0], data2D[:,1],c=x_test)
plt.show()
Этот код работает, если я удаляю c = x_test в последней строке, но цветэто же только один цвет, если я добавлю c=x_test его, скажем, ошибку ValueError: c of shape (444L,) not acceptable as a color sequence for x with size 444, y with size 444
Как исправить код так, чтобы цвет был 6 классов или категорий?