【发布时间】:2014-03-24 16:23:06
【问题描述】:
我正在尝试从 sci-kit learn 运行随机森林分类器,并得到可疑的错误输出 - 不到 1% 的预测是正确的。该模型的表现比偶然性差得多。我对 Python、ML 和 sci-kit 学习(三重打击)相对较新,我担心我缺少一些基本的东西,而不是需要微调参数。我希望有更多老手的眼睛来查看代码,看看设置是否有问题。
我正在尝试根据单词出现来预测电子表格中行的类别 - 因此每行的输入是一个数组,表示每个单词出现的次数,例如[1 0 0 2 0 ... 1]。我正在使用 sci-kit learn 的 CountVectorizer 进行此处理 - 我向它提供包含每行中单词的字符串,并输出单词出现数组。如果此输入由于某种原因不合适,那可能是事情出问题的地方,但我没有在网上或文档中找到任何表明这种情况的信息。
目前,森林的回答正确率约为 0.5%。使用与 SGD 分类器完全相同的输入产生接近 80% 的结果,这表明我正在做的预处理和矢量化很好 - 这是 RF 分类器特有的。我的第一反应是寻找过度拟合,但即使我在 training 数据上运行模型,它仍然几乎所有错误。
我研究过树的数量和训练数据的数量,但这对我来说似乎没有太大变化。我试图只显示相关代码,但如果有帮助可以发布更多。第一个 SO 帖子,感谢所有想法和反馈。
#pull in package to create word occurence vectors for each line
from sklearn.feature_extraction.text import CountVectorizer
vectorizer = CountVectorizer(min_df=1,charset_error='ignore')
X_train = vectorizer.fit_transform(train_file)
#convert to dense array, the required input type for random forest classifier
X_train = X_train.todense()
#pull in random forest classifier and train on data
from sklearn.ensemble import RandomForestClassifier
clf = RandomForestClassifier(n_estimators = 100, compute_importances=True)
clf = clf.fit(X_train, train_targets)
#transform the test data into the vector format
testdata = vectorizer.transform(test_file)
testdata = testdata.todense()
#export
with open('output.csv', 'wb') as csvfile:
spamwriter = csv.writer(csvfile)
for item in clf.predict(testdata):
spamwriter.writerow([item])
【问题讨论】:
-
您是否将电子表格作为 train_file 输入到这一行中的矢量化:X_train = vectorizer.fit_transform(train_file) ?如果是这种情况,那么您的矢量化文件会将您的文件视为单个文本文件并计算该文件中的单词。如果您将输入作为电子表格,那么您不应该使用 CountVectorizer,因为您已经有了字数统计,您应该将该电子表格读入矩阵
-
@Shahram,X_train 本质上是一个字符串列表,每个字符串都包含电子表格中特定行中的所有单词。字符串 1 包含第 1 行中的单词,依此类推。所以我认为矢量化器对于创建单词出现是必要的。我检查了 CountVectorizer 的输出是什么样的,我认为那部分是正确的。不过,我会澄清这一点。
标签: python scikit-learn random-forest