当前位置：首页 > 文章列表 > 文章 > python教程 > TF-IDF差异解析：Scikit-learn实战教程

TF-IDF差异解析：Scikit-learn实战教程

2025-08-01 19:27:29 0浏览收藏

在文本数据处理中，TF-IDF是一种重要的加权技术，用于评估词语在文档中的重要性。Scikit-learn库的`TfidfVectorizer`是实现TF-IDF的强大工具。然而，手动计算TF-IDF值与`TfidfVectorizer`的输出结果常存在差异，这主要源于IDF（逆文档频率）的计算公式不同。本文深入解析了Scikit-learn中`TfidfVectorizer`所采用的标准IDF公式及其优势，例如避免零IDF值和除零错误，并通过实例展示了如何使用`TfidfVectorizer`计算TF-IDF值。同时，强调了理解IDF计算差异对于正确使用`TfidfVectorizer`的关键性，并提供了在实际应用中选择合适TF-IDF计算方式的建议，旨在帮助读者更好地理解和应用TF-IDF技术进行文本向量化。

使用 Scikit-learn 的 TfidfVectorizer 理解 TF-IDF 计算差异

TfidfVectorizer 是 Scikit-learn 库中一个强大的工具，用于将文本数据转换为数值向量，以便机器学习模型可以处理。它通过计算词频-逆文档频率 (TF-IDF) 来实现这一点。然而，当手动计算 TF-IDF 值并与 TfidfVectorizer 的输出进行比较时，可能会发现结果存在差异。这种差异主要源于 IDF (逆文档频率) 的计算方式不同。

理解 TF-IDF 的基本概念

TF-IDF 是一种用于评估单词在文档集合或语料库中重要性的统计方法。它结合了两个关键指标：

TF (词频)： 单词在文档中出现的次数。词频越高，该词对文档越重要。
IDF (逆文档频率)： 衡量一个词在整个语料库中普遍性的指标。如果一个词在很多文档中都出现，则其 IDF 值较低，反之亦然。

TF-IDF 值是将 TF 和 IDF 相乘的结果，用于衡量一个词在特定文档中的重要性，同时考虑了它在整个语料库中的普遍性。

IDF 计算公式的差异

手动计算 TF-IDF 时，常用的 IDF 公式如下（非标准公式）：

IDF(t) = log(N / DF(t))

其中：

N 是语料库中总文档数。
DF(t) 是包含词语 t 的文档数。

然而，Scikit-learn 的 TfidfVectorizer 使用了以下标准 IDF 公式：

IDF(t) = log((1 + N) / (1 + DF(t))) + 1

Scikit-learn IDF 公式的优势

Scikit-learn 采用的公式有以下几个优点：

避免零 IDF 值： 如果一个词出现在所有文档中，DF(t) 将等于 N。在手动计算公式中，这将导致 IDF(t) = log(1) = 0。这会使该词在所有文档中的 TF-IDF 值都为零，从而失去区分度。Scikit-learn 的公式通过添加 1，确保即使词出现在所有文档中，IDF 值也始终大于零。
防止除零错误： 虽然在实际情况中很少发生，但如果语料库中没有包含某个词语，DF(t) 将为零，手动计算公式将导致除零错误。Scikit-learn 的公式通过在分母中添加 1 来避免这种情况。
平滑处理： Scikit-learn 的公式通过添加 1 进行平滑处理，可以减少罕见词语对结果的影响，从而提高模型的泛化能力。

示例说明

考虑以下语料库：

corpus = [
    'This is the first document.',
    'This document is the second document.',
    'And this is the third one.',
    'Is this the first document?',
]

假设我们要计算词语 "document" 的 TF-IDF 值。

N (总文档数)： 4
DF("document") (包含 "document" 的文档数)： 3

手动计算 (非标准公式)：

IDF("document") = log(4 / 3) ≈ 0.2877

Scikit-learn 计算 (标准公式)：

IDF("document") = log((1 + 4) / (1 + 3)) + 1 = log(5 / 4) + 1 ≈ 0.2231 + 1 = 1.2231

可以看到，两个公式计算出的 IDF 值明显不同。

代码示例

以下是如何使用 TfidfVectorizer 计算 TF-IDF 值的示例：

from sklearn.feature_extraction.text import TfidfVectorizer

corpus = [
    'This is the first document.',
    'This document is the second document.',
    'And this is the third one.',
    'Is this the first document?',
]

vectorizer = TfidfVectorizer(norm=None) # 设置 norm=None 以禁用 L2 归一化
tfidf_matrix = vectorizer.fit_transform(corpus)

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

# 打印 TF-IDF 矩阵
import pandas as pd
df = pd.DataFrame(tfidf_matrix.toarray(), columns = feature_names)
print(df)

# 输出 'document' 的 IDF 值
print(f"IDF('document'): {vectorizer.idf_[vectorizer.vocabulary_['document']]}")

输出结果：

      and  document  first  is  one  second  the  third  this
0   0.000000  1.000000   1.693147  1.0  1.693147    0.000000  1.0  0.000000   1.0
1   0.000000  2.000000   0.000000  1.0  0.000000    1.693147  1.0  0.000000   1.0
2   1.693147  0.000000   0.000000  1.0  1.693147    0.000000  1.0  1.693147   1.0
3   0.000000  1.000000   1.693147  1.0  0.000000    0.000000  1.0  0.000000   1.0
IDF('document'): 1.2231435513142097

注意事项和总结