Python:管道使用第一个分类器的一些结果到第二个分类器(sklearn)

时间:2018-02-22 10:40:01

标签: python scikit-learn pipeline text-classification

我想使用GaussianNB分类为A / B类,然后使用MultiNomialNB将类型A分类为子类别a1 / a2 / a3

我的问题是如何将另一个第一个分类器插入管道并使用 仅输入第二个分类器的结果?

我现在拥有的东西:

pipeline1 = Pipeline([
('bow', CountVectorizer(analyzer=preprocess)), 
('tfidf', TfidfTransformer()),  
('classifier', MultinomialNB()),])

scores = cross_val_score(pipeline1,train_X,train_Y,cv=cv,scoring=scorer)

pipeline1.fit(train_X,train_Y)
predictions= pipeline1.predict(test_X)
print (classification_report(predictions,test_Y))

我的模拟数据集

的示例

enter image description here

0 个答案:

没有答案