美文网首页
XGBoost_调试方法

XGBoost_调试方法

作者: xieyan0811 | 来源:发表于2018-10-06 09:57 被阅读64次

    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_调试方法

          本文链接:https://www.haomeiwen.com/subject/mlocaftx.html