Python:jupyter 模型可视化（VS）

是清宵呀2024-08-13 9:16

step1:打开vs安装扩展安装后重新启动vs

建立可视化模型

复制代码

import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn import tree

music_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语言即可刷新