如何获得机器学习模型(尤其是XG boost)的模型摘要,如下面的屏幕截图所示?

时间:2019-03-05 15:16:41

标签: python scikit-learn xgboost

我已经在数据集上运行了Xgboost回归器,并且我需要模型摘要(例如R平方,峰度等),类似于所附的屏幕截图。任何帮助将不胜感激。

Screenshot

1 个答案:

答案 0 :(得分:1)

您可以使用sklearn实现获取R square valueperformance metrics

但是诸如co-eff之类的其他内容与Xgboost模型无关,因为它是一个非线性模型。因此,您无法根据Xgboost模型为每个功能分配单个coeff。

有更好的方法来理解模型的可解释性,我建议的方法之一是[SHAP Tree explainers][2]

Blog了解shap的工作原理。