【发布时间】:2014-11-28 17:04:46
【问题描述】:
假设我有一些要使用 kmeans 聚类的文本句子。
sentences = [
"fix grammatical or spelling errors",
"clarify meaning without changing it",
"correct minor mistakes",
"add related resources or links",
"always respect the original author"
]
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.cluster import KMeans
vectorizer = CountVectorizer(min_df=1)
X = vectorizer.fit_transform(sentences)
num_clusters = 2
km = KMeans(n_clusters=num_clusters, init='random', n_init=1,verbose=1)
km.fit(X)
现在我可以预测新文本将属于哪个类,
new_text = "hello world"
vec = vectorizer.transform([new_text])
print km.predict(vec)[0]
但是,假设我应用 PCA 将 10,000 个功能减少到 50 个。
from sklearn.decomposition import RandomizedPCA
pca = RandomizedPCA(n_components=50,whiten=True)
X2 = pca.fit_transform(X)
km.fit(X2)
我不能再做同样的事情来预测新文本的集群,因为矢量化器的结果不再相关
new_text = "hello world"
vec = vectorizer.transform([new_text]) ##
print km.predict(vec)[0]
ValueError: Incorrect number of features. Got 10000 features, expected 50
那么如何将我的新文本转换为低维特征空间?
【问题讨论】:
标签: python machine-learning scikit-learn pca