XGBoost_调试方法

1. 调试test目录下的测试用例

 在测试程序后面加入以下代码,即可启动调试程序

if __name__ == "__main__":
    unittest.main()

2. 显示树结构

import matplotlib.pyplot as plt 
fig,ax = plt.subplots()
xgb.plot_tree(gbdt_03a, ax = ax, num_trees=0) # 显示模型中的第一棵树
plt.show()

3. 打印详细调试信息

 在xgb的params中设置:

'silent': 0,
'debug_verbose': 5

4. 修改源码

 修改c++源码后如果运行c++程序,在xgboost目录下执行编译命令make,重新生成二进制程序xgboost,运行即可。

 修改c++源码后如果运行Python程序,需要将xgboost/lib/libxgboost.so,复制到python对应的库目录下(如:/usr/local/lib/python2.7/dist-packages/xgboost/lib/),注意源码和python库的版本一致性。

你可能感兴趣的:(XGBoost_调试方法)