Python:jupyter 模型可视化(VS)

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

相关推荐
兔子的洋葱圈7 分钟前
【django】1-2 django项目的请求处理流程(详细)
后端·python·django
独好紫罗兰15 分钟前
洛谷题单3-P5719 【深基4.例3】分类平均-python-流程图重构
开发语言·python·算法
276695829217 分钟前
美团民宿 mtgsig 小程序 mtgsig1.2 分析
java·python·小程序·美团·mtgsig·mtgsig1.2·美团民宿
橘子在努力25 分钟前
【橘子大模型】关于PromptTemplate
python·ai·llama
SheepMeMe43 分钟前
蓝桥杯2024省赛PythonB组——日期问题
python·算法·蓝桥杯
莓事哒1 小时前
selenium和pytessarct提取古诗文网的验证码(python爬虫)
爬虫·python·selenium·测试工具·pycharm
q567315231 小时前
使用puppeteer库编写的爬虫程序
爬虫·python·网络协议·http
mosquito_lover11 小时前
Python数据分析与可视化实战
python·数据挖掘·数据分析
eqwaak02 小时前
量子计算与AI音乐——解锁无限可能的音色宇宙
人工智能·爬虫·python·自动化·量子计算
SylviaW082 小时前
python-leetcode 63.搜索二维矩阵
python·leetcode·矩阵