python - 尝试使用文本分类并接收错误代码
问题描述
输入的代码:
from sklearn.feature_extraction.text import CountVectorizer
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(twenty_train.data)
X_train_counts.shape
from sklearn.feature_extraction.text import TfidfTransformer
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
X_train_tfidf.shape
错误信息:
NameError: name 'tfidf_transformer' is not defined
我对此很陌生,所以请多多包涵。
为什么我会收到此错误消息?我已经定义tfidf_transformer
了不是吗?
解决方案
使用: TfidfTransformer().fit(X_train_counts) 而不是 tfidf_transformer.fit_transform(X_train_counts)