xgboost中plot_tree()內部是依賴於graphviz插件,所以,如果想用xgboost訓練數據,並將訓練結果以圖的形式展示處理,需要graphviz插件。我建議大家一併安裝了。
其實,安裝xgboost和graphviz的過程不難,只是對於初學者來說有點繁瑣,所以我把裝了一上午的經驗總結一下,希望能給大家提供一些幫助。
前提條件是系統安裝了Anaconda2.7/3.6
如果沒有安裝,則到https://www.anaconda.com/download/ 去下載Anaconda安裝包進行安裝1、xgboost安裝說明:
①Linux系統下安裝xgboost:
在要安裝xgboost的目錄下(我的是:~/anaconda3/lib),
右鍵點擊open Terminal
輸入命令:pip install xgboost
安裝成功即可看到 Successfully installed xgboost-0.71
②Windows系統下安裝xgboost:
可以參考這個網站進行安裝: https://blog.csdn.net/leo_xu06/article/details/52300869
也可以簡單一點,直接用別人編譯好的文件進行安裝,下載地址如下:
Anaconda2.7 點擊這個下載:https://pan.baidu.com/s/11fEB86gOjQRIvb9AQgzYHA
Anaconda3.6 點擊這個下載:https://pan.baidu.com/s/1sg_xjusNO7nEgUId8bOAqQ
下載最新版本的 xgboost.dll 放到 xgboost-master\python-package\xgboost 下,否則會報錯
執行命令:
先進入xgboost-master\python-packag
再然後:python setup.py install
這樣就完成了xgboost的安裝。
2、graphviz安裝說明:
①Linux系統下安裝graphviz:
執行命令:sudo apt-get install graphviz
安裝完成 。
②Windows下安裝graphviz:
下載 graphviz.msi , 地址:https://pan.baidu.com/s/1JoE5Al3o7YqztIs_o8u22g ,一步一步安裝
配置環境變量,將graphviz路徑配置到path環境變量中。
在cmd中輸入命令:pip install graphviz
安裝完成。
備註:如果在電腦中安裝了多個版本的Anaconda,要注意在pyCharm中使用的Anaconda版本要和安裝graphviz的Anaconda版本一致。
即:
和:
graphviz安裝到哪裏與path路徑中配了哪個版本的Anaconda有關係,可以在path中修改進行控制