首页 文章

Python决策树图像sklearn

提问于
浏览
-1

我正在使用棒球数据集,其中每一行都是一个单独的音高 . 每列是关于该音调的属性 . 示例1俯仰可以是95英里/小时,2000转/分的旋转速度,100英里/小时的蝙蝠的退出速度等 .

我从下面这个网页的代码中 Build 了一个决策树,并使用俯仰速度和旋转速率来预测该音调是否导致命中 . 我的准确率达到了81%......但谁在乎呢?我需要能够从决策树中获得一些见解 . 例如,我需要能够读取类似的东西,超过95英里/小时的旋转速度超过3,000,导致87%的时间“不受打击” .

如何绘制决策树的图像以查看我可以提出建议的此类数据?

http://dataaspirant.com/2017/02/01/decision-tree-algorithm-python-with-scikit-learn/

1 回答

  • 0

    使用代码训练到文件后转储决策树模型

    with open("fine_name.dot", 'w') as f:
        f = tree.export_graphviz(tree_name, out_file=f, feature_names=columns)
    

    复制生成的文件内容并粘贴到http://webgraphviz.com/的编辑器中,以显示决策树 . 它将决策树转换为白色框:D

相关问题