【发布时间】:2015-03-30 22:45:25
【问题描述】:
我可以使用 scikit-learn 使用以下代码打印分类报告:
def predict_and_report_test(self, prediction_model):
print(prediction_model.algorithm + ' Test')
prediction_model.model = prediction_model.model.fit(self.X_train, self.y_train).predict(self.X_test)
print(classification_report(self.y_test, prediction_model.model, target_names=None))
这段代码给了我以下输出:
问题是分类报告以字符串形式返回此信息,所以我的问题是:scikit learn 是否提供了任何简单的方法来访问每个类的支持并可能将它们存储在一个数组中,这样它们就可以用来绘制每个类的支持度图?
【问题讨论】:
标签: machine-learning scikit-learn