step1:打开vs安装扩展 安装后重新启动vs
建立可视化模型
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn import treemusic_data = pd.read_csv('music.csv')X=music_data.drop(columns=['genre'])
Y=music_data['genre']model=DecisionTreeClassifier()
model.fit(X,Y)# Y.unique() 去重
tree.export_graphviz(model,out_file='music-rec.dot',feature_names=['age','gender'],class_names=sorted(Y.unique()),label='all',rounded=True,filled=True)
运行后桌面(当前编写代码的目录)会有相应文件
用vs打开.dot 文件
点击右边那个dot图标即可
如果没有,直接新建一个file,选择dot语言即可刷新