【发布时间】:2020-09-16 03:57:15
【问题描述】:
我需要可视化 XGBoost 的效果,并且我想从中提取一棵在测试集上具有最高准确度的树。是以任何方式内置的,还是我必须手动测试所有的树?我使用的是 Python 版本。
【问题讨论】:
-
请提供更多详细信息。
-
我有 XGBoost 模型。我想提取一棵树。该树必须具有测试集上所有树的最高准确度。我必须一一检查所有树木,还是可以更轻松?
-
你使用的是哪个框架?
-
Python 的官方 XGBoost 实现
-
我认为您必须手动检查它们:model.get_booster().get_dump()
标签: python decision-tree xgboost