python-3.x - 腌制随机森林模型的混淆矩阵

Question

是否可以仅使用保存为 pickle-dump 的模型来重建袋外验证的完整混淆矩阵？

mymodel.oob_score_报告 OOB 分数为 0.75682

我在这里发现可以像这样提取这个模型的预测标签：

pred_train = numpy.argmax(mymodel.oob_decision_function_,axis=1)

但是火车标签是否也保存在模型文件中的某个地方？

标签： python-3.xscikit-learnpicklerandom-forestconfusion-matrix

Answer 1

这取决于您使用 导出的内容pickle.dump()，但一般来说，如果您只导出随机森林，则不会。

sklearn.ensemble.RandomForestClassifier只有这些属性，而基本事实不是其中之一。

oob_decision_functionn_samples仅存储大小为 ( , )的数组，n_classes因此您无法从那里获取它们。没有其他属性可以存储 oob 基本事实。