去除决策树右侧的小黑块

在解决中文乱码问题的时候,发现结果图中右侧有一个小黑块

去除决策树右侧的小黑块_第1张图片

代码如下:

#解决中文乱码问题
f1_tree_val = f1_tree.getvalue()
dot_data_val = f1_tree_val.replace('helvetica', 'Microsoft YaHei')
graph = pydotplus.graph_from_dot_data(dot_data_val)
graph.write_png('f1_tree.png')
Image(filename='f1_tree.png')

如何解决这个问题呢?

 只需在第二行后面加一句

dot_data_val = dot_data_val.replace('\n','')

问题得到解决

你可能感兴趣的:(决策树,算法)