在保存的分类器中进行预测



我使用TfIdfVectorizer和MultinomialNB来训练我的模型,并将其保存到pickle文件中。

现在我试图使用来自另一个文件的分类器来预测看不见的数据,我不能这样做,因为它告诉我分类器的特征数量与我当前语料库的特征数量不相同。

这是我试图预测的代码。函数do_vectorize与训练中使用的完全相同。

def do_vectorize(data, stop_words=[], tokenizer_fn=tokenize):
    vectorizer = TfidfVectorizer(stop_words=stop_words, tokenizer=tokenizer_fn)
    X = vectorizer.fit_transform(data)
    return X, vectorizer
# Vectorizing the unseen documents 
matrix, vectorizer = do_vectorize(corpus, stop_words=stop_words)
# Predicting on the trained model
clf = pickle.load(open('../data/classifier_0.5_function.pkl', 'rb'))
predictions = clf.predict(matrix)

然而,我收到的错误是特征的数量不同

ValueError: Expected input with 65264 features, got 472546 instead

这意味着我还必须从训练中节省词汇量以进行测试?如果有培训中不存在的条款怎么办?

我尝试使用scikit-learn中的管道,具有相同的矢量器和分类器,并且两者的参数相同。但是,从1个小时变成6个多小时太慢了,所以我更喜欢手动操作。

这意味着我还必须从训练中节省词汇量以进行测试?

是的,你必须保存整个tfidf矢量器,这特别意味着保存词汇。

如果有培训中不存在的术语会发生什么?

它们将被忽略,这是完全有意义的,因为您没有关于此的训练数据,因此没有什么需要考虑的(有更复杂的方法仍然可以使用它,但它们不使用像tfidf这样简单的方法)。

我尝试使用scikit-learn中的管道,具有相同的矢量器和分类器,并且两者的参数相同。但是,从1个小时转到6个多小时太慢了,所以我更喜欢手动操作。

在使用管道时应该很少甚至没有开销,但是只要您还记得存储矢量器,手动操作也是可以的。

您必须在使用tfidf矢量器时分配最大特征限制这样的

tfidf_vectorizer = TfidfVectorizer(max_features = 1200)

,然后使用相同的功能限制将测试数据转换为

相关内容

  • 没有找到相关文章

最新更新