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

相关推荐
__lost6 分钟前
Python图像变清晰与锐化,调整对比度,高斯滤波除躁,卷积锐化,中值滤波钝化,神经网络变清晰
python·opencv·计算机视觉
海绵波波10712 分钟前
玉米产量遥感估产系统的开发实践(持续迭代与更新)
python·flask
逢生博客1 小时前
使用 Python 项目管理工具 uv 快速创建 MCP 服务(Cherry Studio、Trae 添加 MCP 服务)
python·sqlite·uv·deepseek·trae·cherry studio·mcp服务
堕落似梦1 小时前
Pydantic增强SQLALchemy序列化(FastAPI直接输出SQLALchemy查询集)
python
坐吃山猪2 小时前
Python-Agent调用多个Server-FastAPI版本
开发语言·python·fastapi
Bruce-li__2 小时前
使用Django REST Framework快速开发API接口
python·django·sqlite
小兜全糖(xdqt)2 小时前
python 脚本引用django中的数据库model
python·django
Arenaschi3 小时前
SQLite 是什么?
开发语言·网络·python·网络协议·tcp/ip
纪元A梦3 小时前
华为OD机试真题——推荐多样性(2025A卷:200分)Java/python/JavaScript/C++/C语言/GO六种最佳实现
java·javascript·c++·python·华为od·go·华为od机试题
仙人掌_lz3 小时前
人工智能与机器学习:Python从零实现性回归模型
人工智能·python·机器学习·线性回归