k 均值聚类中如何使用 tfidf 值

Question

我正在使用 sckit-learn 库将 K-means 聚类与 TF-IDF 结合使用。我知道 K-means 使用距离来创建聚类，距离用（x 轴值，y 轴值）表示，但 tf-idf 是单个数值。我的问题是这个tf-idf值是如何通过K-means聚类转换成(x,y)值的。

Answer 1

TF-IDF 不是单个值（即标量）。对于每个文档，它 returns 一个向量，其中向量中的每个值对应于词汇表中的每个单词。

from sklearn.feature_extraction.text import TfidfVectorizer
import numpy as np
from scipy.sparse.csr import csr_matrix

sent1 = "the quick brown fox jumps over the lazy brown dog"
sent2 = "mr brown jumps over the lazy fox"

corpus = [sent1, sent2]
vectorizer = TfidfVectorizer(input=corpus)

X = vectorizer.fit_transform(corpus)
print(X.todense())

[输出]:

matrix([[0.50077266, 0.35190925, 0.25038633, 0.25038633, 0.25038633,
         0.        , 0.25038633, 0.35190925, 0.50077266],
        [0.35409974, 0.        , 0.35409974, 0.35409974, 0.35409974,
         0.49767483, 0.35409974, 0.        , 0.35409974]])

它 returns 一个二维矩阵，其中行代表句子，列代表词汇。

>>> vectorizer.vocabulary_
{'the': 8,
 'quick': 7,
 'brown': 0,
 'fox': 2,
 'jumps': 3,
 'over': 6,
 'lazy': 4,
 'dog': 1,
 'mr': 5}

因此，当 K-means 试图找到两个文档之间的 distance/similarity 时，它正在执行矩阵中两行之间的相似性。例如。假设相似性只是两行之间的点积：

import numpy as np
vector1 = X.todense()[0]
vector2 = X.todense()[1]
float(np.dot(vector1, vector2.T))

[输出]:

0.7092938737640962

Chris Potts 有一个很好的教程，介绍如何创建矢量 space 模型，例如 TF-IDF 模型 http://web.stanford.edu/class/linguist236/materials/ling236-handout-05-09-vsm.pdf

k 均值聚类中如何使用 tfidf 值

how tfidf value is used in k-means clustering

python-3.x

nlp

k-means

tf-idf

tfidfvectorizer