决策树算法的原理,对于研究机器学习的同学想必都非常的熟悉。
在进行数据分析过程中,我们想进一步深入的了解数据特征重要性等等时,最好能够通过数据可视化的方式深入的了解每一个特征对于数据分类划分的重要性。
这时,我们学习《机器学习实战:基于Scikit-Learn和TensorFlow》的案例,引入sklearn模块中的graphviz
模块
首先,构建存储图片位置的路径
import os
PROJECT_ROOT_DIR = "."
CHAPTER_ID = "decision_trees"
IMAGES_PATH = os.path.join(PROJECT_ROOT_DIR, "images", CHAPTER_ID)
os.