【发布时间】:2019-02-06 22:24:00
【问题描述】:
我是第一次玩 sklearn 和 NLP,并认为我理解我所做的一切,直到我不知道如何修复这个错误。以下是相关代码(大部分改编自http://zacstewart.com/2015/04/28/document-classification-with-scikit-learn.html):
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.pipeline import Pipeline, FeatureUnion
from sklearn.preprocessing import StandardScaler
from sklearn.decomposition import TruncatedSVD
from sgboost import XGBClassifier
from pandas import DataFrame
def read_files(path):
for article in os.listdir(path):
with open(os.path.join(path, doc)) as f:
text = f.read()
yield os.path.join(path, article), text
def build_data_frame(path, classification)
rows = []
index = []
for filename, text in read_files(path):
rows.append({'text': text, 'class': classification})
index.append(filename)
df = DataFrame(rows, index=index)
return df
data = DataFrame({'text': [], 'class': []})
for path, classification in SOURCES: # SOURCES is a list of tuples
data = data.append(build_data_frame(path, classification))
data = data.reindex(np.random.permutation(data.index))
classifier = Pipeline([
('features', FeatureUnion([
('text', Pipeline([
('tfidf', TfidfVectorizer()),
('svd', TruncatedSVD(algorithm='randomized', n_components=300)
])),
('words', Pipeline([('wscaler', StandardScaler())])),
])),
('clf, XGBClassifier(silent=False)),
])
classifier.fit(data['text'].values, data['class'].values)
加载到 DataFrame 中的数据是经过预处理的文本,其中包含所有停用词、标点符号、unicode、大写字母等。这是我在分类器上调用 fit 时遇到的错误,其中 ... 表示应该在管道中进行 vecorized 的文档之一:
ValueError: could not convert string to float: ...
我一开始以为 TfidfVectorizer() 不起作用,导致 SVD 算法出错,但是在我从管道中提取每个步骤并按顺序执行它们之后,同样的错误只出现在 XGBClassifer.fit() 上。
让我更困惑的是,我试图在解释器中逐步拆分这个脚本,但是当我尝试导入 read_files 或 build_data_frame 时,同样的 ValueError 出现了我的一个字符串,但这是仅仅在:
from classifier import read_files
我不知道这是怎么发生的,如果有人知道我的明显错误可能是什么,我将不胜感激。试图独自思考这些概念,但遇到这样的问题让我感到非常无能为力。
【问题讨论】:
标签: python scikit-learn nlp text-classification