python 实现决策树规则

python 实现决策树规则

sql 复制代码
----决策树探索----
from sklearn import tree
import graphviz
import pydotplus
from pydotplus.graphviz import graph_from_dot_data
from sklearn.tree import export_graphviz
import os   
os.environ["PATH"] += os.pathsep + 'D:/软件安装/windows_10_msbuild_Release_graphviz-9.0.0-win32/Graphviz/bin/'

x = bj_test_union[['x1','x2','x3','x4']]
y = bj_test_union["dpd30"]
cart_tree = tree.DecisionTreeRegressor(max_depth=2)
cart_tree = cart_tree.fit(x,y)


cart_tree_data = tree.export_graphviz(
cart_tree,
out_file =None,
proportion=True,
feature_names=['x1','x2','x3','x4'],
class_names=["0","1"],
filled=True,
rounded=True,
special_characters=True)
graph = graphviz.Source(cart_tree_data)
graph
相关推荐
Jay Kay35 分钟前
TensorFlow源码深度阅读指南
人工智能·python·tensorflow
会的全对٩(ˊᗜˋ*)و1 小时前
【数据挖掘】数据挖掘综合案例—银行精准营销
人工智能·经验分享·python·数据挖掘
___波子 Pro Max.1 小时前
GitHub Actions配置python flake8和black
python·black·flake8
阿蒙Amon2 小时前
【Python小工具】使用 OpenCV 获取视频时长的详细指南
python·opencv·音视频
橘子编程3 小时前
Python-Word文档、PPT、PDF以及Pillow处理图像详解
开发语言·python
蓝婷儿3 小时前
Python 机器学习核心入门与实战进阶 Day 2 - KNN(K-近邻算法)分类实战与调参
python·机器学习·近邻算法
之歆4 小时前
Python-封装和解构-set及操作-字典及操作-解析式生成器-内建函数迭代器-学习笔记
笔记·python·学习
天天爱吃肉82184 小时前
ZigBee通信技术全解析:从协议栈到底层实现,全方位解读物联网核心无线技术
python·嵌入式硬件·物联网·servlet
Allen_LVyingbo5 小时前
Python常用医疗AI库以及案例解析(2025年版、上)
开发语言·人工智能·python·学习·健康医疗
智能砖头5 小时前
LangChain 与 LlamaIndex 深度对比与选型指南
人工智能·python