我能够使用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))
这段代码会产生以下输出:
问题在于classification_report返回的信息是一个字符串,所以我的问题是:scikit-learn是否提供了简单的方法来访问每个类别的支持量,并可能将它们存储在一个数组中,以便用于绘制每个类别的支持量图表?
回答:
有一个名为precision_recall_fscore_support的函数,这是classification_report调用的函数: http://scikit-learn.org/dev/modules/generated/sklearn.metrics.precision_recall_fscore_support.html#sklearn.metrics.precision_recall_fscore_support