理解并正确使用 TfidfVectorizer 计算 TF-IDF 值

本文旨在帮助读者理解 TfidfVectorizer 在 scikit-learn 中计算 TF-IDF 值的原理，特别是关于 IDF 的计算方式，以及如何通过调整 smooth_idf 参数来影响计算结果。同时，澄清了 TF 的计算方式，避免混淆。通过本文，读者可以更准确地使用 TfidfVectorizer 进行文本特征提取。

TfidfVectorizer 是 scikit-learn 库中一个强大的文本特征提取工具，它将文本数据转换为数值型特征，以便机器学习模型能够处理。TF-IDF (Term Frequency-Inverse Document Frequency) 是一种常用的文本特征权重计算方法，它结合了词频 (TF) 和逆文档频率 (IDF) 来衡量一个词在文档集中的重要性。理解 TfidfVectorizer 的内部计算过程对于有效地使用它至关重要。

IDF 的计算

在 TfidfVectorizer 中，IDF 的计算公式默认包含一个平滑项 smooth_idf。 默认情况下，smooth_idf 设置为 True，这会对 IDF 的计算产生影响。

当 smooth_idf = True 时：

IDF 的计算公式为：

算家云

高效、便捷的人工智能算力服务平台

37

查看详情

IDF(t) = ln((1 + n) / (1 + df(t))) + 1

其中：

• n 是文档总数。
• df(t) 是包含词语 t 的文档数量。
• ln 是自然对数。

smooth_idf 的作用是防止在计算 IDF 时出现除以零的情况，并平滑 IDF 值，使得罕见词语的 IDF 值不会过高。

当 smooth_idf = False 时：

IDF 的计算公式为：

IDF(t) = ln(n / df(t)) + 1

可以看到，当 smooth_idf 为 False 时，公式中分子和分母都不加 1。

示例：

假设我们有以下三个文档：

data = ['Souvenir shop|Architecture and art|Culture and history',
        'Souvenir shop|Resort|Diverse cuisine|Fishing|Shop games|Beautiful scenery',
        'Diverse cuisine|Resort|Beautiful scenery']

要使用 TfidfVectorizer 计算这些文档的 TF-IDF 值，可以执行以下操作：

from sklearn.feature_extraction.text import TfidfVectorizer
import numpy as np

data = ['Souvenir shop|Architecture and art|Culture and history',
        'Souvenir shop|Resort|Diverse cuisine|Fishing|Shop games|Beautiful scenery',
        'Diverse cuisine|Resort|Beautiful scenery']

vectorizer = TfidfVectorizer()
tfidf_matrix = vectorizer.fit_transform(data)

# 获取词汇表
feature_names = vectorizer.get_feature_names_out()

# 获取 IDF 值
idf_values = vectorizer.idf_

# 将词汇和 IDF 值对应起来
word_idf = dict(zip(feature_names, idf_values))

# 打印 "art" 的 IDF 值
print(f"IDF of 'art': {word_idf['art']}")

# 关闭 smooth_idf 再次计算
vectorizer = TfidfVectorizer(smooth_idf=False)
tfidf_matrix = vectorizer.fit_transform(data)
idf_values = vectorizer.idf_
feature_names = vectorizer.get_feature_names_out()
word_idf = dict(zip(feature_names, idf_values))
print(f"IDF of 'art' (smooth_idf=False): {word_idf['art']}")

在这个例子中，文档总数 n 为 3。 词语 "art" 只出现在第一个文档中，因此 df("art") 为 1。

• 当 smooth_idf=True (默认) 时， art 的 IDF 值为 ln((3 + 1) / (1 + 1)) + 1 = ln(2) + 1 ≈ 1.6931
• 当 smooth_idf=False 时， art 的 IDF 值为 ln(3 / 1) + 1 = ln(3) + 1 ≈ 2.0986

TF 的计算

TF (Term Frequency) 表示词语在文档中出现的频率。 在 TfidfVectorizer 中，TF 只是词语在文档中出现的次数，没有进行文档长度的归一化。 归一化步骤发生在乘以 IDF 之后。

例如，在文档 "Souvenir shop|Architecture and art|Culture and history" 中，词语 "art" 出现了一次，所以 TF("art") = 1。

TF-IDF 的计算流程总结

1. 词频 (TF) 计算： 统计每个词语在每个文档中出现的次数。
2. 逆文档频率 (IDF) 计算： 根据文档总数和包含词语的文档数，计算每个词语的 IDF 值。 可以使用 smooth_idf 参数来控制是否进行平滑处理。
3. TF-IDF 计算： 将 TF 和 IDF 相乘，得到每个词语在每个文档中的 TF-IDF 值。
4. 归一化 (Normalization): TfidfVectorizer 默认会对每个文档的 TF-IDF 向量进行 L2 归一化，确保每个向量的欧几里得范数为 1。 可以通过设置 norm 参数来控制是否进行归一化。

注意事项

• TfidfVectorizer 默认会对文本进行预处理，例如转换为小写、去除停用词等。 可以通过调整 lowercase、stop_words 等参数来控制预处理过程。
• 理解 TfidfVectorizer 的参数对于获得最佳的特征提取效果至关重要。 仔细阅读 scikit-learn 的官方文档，了解每个参数的作用和用法。

总结

TfidfVectorizer 是一个功能强大的文本特征提取工具，通过理解其内部的 TF-IDF 计算过程，可以更好地利用它来构建有效的机器学习模型。 特别是需要注意 smooth_idf 参数对 IDF 计算的影响，以及 TF 的计算方式。通过合理地调整参数，可以获得更符合实际需求的文本特征表示。

以上就是理解并正确使用 TfidfVectorizer 计算 TF-IDF 值的详细内容，更多请关注php中文网其它相关文章！