用于XGBoost模型的解释方法
基本信息
申请号 | CN202010439751.9 | 申请日 | - |
公开(公告)号 | CN111612156A | 公开(公告)日 | 2020-09-01 |
申请公布号 | CN111612156A | 申请公布日 | 2020-09-01 |
分类号 | G06N20/00(2019.01)I | 分类 | 计算;推算;计数; |
发明人 | 袁秋龙;周珊;江文斌;李健 | 申请(专利权)人 | 上海携程国际旅行社有限公司 |
代理机构 | 上海弼兴律师事务所 | 代理人 | 薛琦;张冉 |
地址 | 200335上海市长宁区金钟路968号16号楼10F | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种用于XGBoost模型的解释方法,包括以下步骤:对XGBoost模型进行树模型元结构解析,以解析出每个单棵树的树结构;对所述XGBoost模型输入测试样本,根据树结构获取所述测试样本对应的有效叶子节点以及有效叶子节点的树的有效路径;根据有效路径计算出特征的贡献值;以及,根据获取到的贡献值对所述XGBoost模型进行解释。本发明根据XGBoost模型的结构信息,同时利用树型结构的路径来有效地对XGBoost模型的判别结果进行解释。 |
