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

相关推荐
Q_Q51100828527 分钟前
python+django/flask婚纱摄影拍照管理系统
spring boot·python·django·flask·node.js·php
可触的未来,发芽的智生1 小时前
触摸未来2025-10-18:生成文字的小宇宙矩阵溯源
人工智能·python·神经网络·程序人生·自然语言处理
yudiandian20141 小时前
03 Eclipse 配置 JDK 环境
java·ide·eclipse
_码力全开_1 小时前
P1005 [NOIP 2007 提高组] 矩阵取数游戏
java·c语言·c++·python·算法·矩阵·go
jerryinwuhan1 小时前
机器人模拟器(python)
开发语言·python·机器人
AhriProGramming1 小时前
Flask-SQLAlchemy精读-双语精选文章
python·算法·flask
列兵阿甘1 小时前
知微传感Dkam系列3D相机SDK例程篇:Python获取内外参
python·数码相机·3d
Full Stack Developme2 小时前
java.net 包详解
java·python·.net
彦楠2 小时前
IDEA实用快捷键
java·ide·intellij-idea
江太翁3 小时前
Kotlin 与 Java 互操作中常用注解
java·python·kotlin