函数：tree_plot

最新推荐文章于 2024-11-12 17:52:36 发布

苏打水的杯子

最新推荐文章于 2024-11-12 17:52:36 发布

阅读量4.8k

点赞数

CC 4.0 BY-SA版权

分类专栏：人工智能文章标签：决策树

本文链接：https://blog.csdn.net/qq_38784454/article/details/79730545

人工智能专栏收录该内容

82 篇文章 ¥59.90 ¥99.00

订阅专栏

本文介绍了一个名为tree_plot的Matlab函数，该函数用于绘制决策树。通过参考treeplot函数，它生成节点布局并用红色圆点表示节点，红线连接节点。函数接受参数p和nodevalues，支持自定义节点标签。对于节点数量小于500的情况，函数会在图上显示每个节点的编号，否则仅绘制连线。最后，函数设置x轴标签为高度并调整坐标轴范围。

function tree_plot( p ,nodevalues)
%% 参考treeplot函数

[x,y,h]=treelayout(p);
f = find(p~=0);
pp = p(f);
X = [x(f); x(pp); NaN(size(f))];
Y = [y(f); y(pp); NaN(size(f))];

X = X(:);
Y = Y(:);

    n = length(p);
    if n < 500,
        hold on ; 
        plot (x, y, 'ro', X, Y, 'r-');
        nodesize = length(x);
        for i=1:nodesize
%            text(x(i)+0.01,y(i),['node' num2str(i)]); 
            text(x(i)+0.01,y(i),nodevalues{1,i}); 
        end
        hold off;
    else
        plot (X, Y, 'r-');
    end;

xlabel(['height = ' int2str(h)]);
axis([0 1 0 1]);

end