ValueError: Invalid decision type in tree model.

使用LightGBM模型可視化時(shí)報(bào)錯(cuò)ValueError: Invalid decision type in tree model.

lgb.create_tree_digraph()
lgb.plot_tree()

錯(cuò)誤定位到python-package/lightgbm/plotting.py中的271行
原因是root['decision_type']為'<='，而判斷條件錯(cuò)誤，微軟官方LightGBM好像已經(jīng)修正了錯(cuò)誤，但可能是我安裝版本的問題
解決方法為進(jìn)入/.local/lib/python3.6/site-packages/lightgbm目錄
管理員權(quán)限打開文件sudo gedit plotting.py
相關(guān)位置修改為如下代碼

                if info in {'split_gain', 'internal_value', 'internal_count'}:
                    label += '\n' + info + ':' + str(root[info])
            graph.node(name, label=label)
            # if root['decision_type'] == 'no_greater':
            if root['decision_type'] == '<=':
                l_dec, r_dec = '<=', '>'
            # elif root['decision_type'] == 'is':
            elif root['decision_type'] == '=':

完美解決
參考：https://github.com/microsoft/LightGBM/pull/810/files/7e824df64a739b6818f2d10bf3dbdc2cfd7fe876

\color{red}{(原創(chuàng),轉(zhuǎn)載請(qǐng)注明來源)}

最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容