用于XGBoost模型的解釋方法
基本信息
申請(qǐng)?zhí)?/td> | CN202010439751.9 | 申請(qǐng)日 | - |
公開(kāi)(公告)號(hào) | CN111612156A | 公開(kāi)(公告)日 | 2020-09-01 |
申請(qǐng)公布號(hào) | CN111612156A | 申請(qǐng)公布日 | 2020-09-01 |
分類號(hào) | G06N20/00(2019.01)I | 分類 | 計(jì)算;推算;計(jì)數(shù); |
發(fā)明人 | 袁秋龍;周珊;江文斌;李健 | 申請(qǐng)(專利權(quán))人 | 上海攜程國(guó)際旅行社有限公司 |
代理機(jī)構(gòu) | 上海弼興律師事務(wù)所 | 代理人 | 薛琦;張冉 |
地址 | 200335上海市長(zhǎng)寧區(qū)金鐘路968號(hào)16號(hào)樓10F | ||
法律狀態(tài) | - |
摘要
摘要 | 本發(fā)明公開(kāi)了一種用于XGBoost模型的解釋方法,包括以下步驟:對(duì)XGBoost模型進(jìn)行樹(shù)模型元結(jié)構(gòu)解析,以解析出每個(gè)單棵樹(shù)的樹(shù)結(jié)構(gòu);對(duì)所述XGBoost模型輸入測(cè)試樣本,根據(jù)樹(shù)結(jié)構(gòu)獲取所述測(cè)試樣本對(duì)應(yīng)的有效葉子節(jié)點(diǎn)以及有效葉子節(jié)點(diǎn)的樹(shù)的有效路徑;根據(jù)有效路徑計(jì)算出特征的貢獻(xiàn)值;以及,根據(jù)獲取到的貢獻(xiàn)值對(duì)所述XGBoost模型進(jìn)行解釋。本發(fā)明根據(jù)XGBoost模型的結(jié)構(gòu)信息,同時(shí)利用樹(shù)型結(jié)構(gòu)的路徑來(lái)有效地對(duì)XGBoost模型的判別結(jié)果進(jìn)行解釋。?? |
