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
相关推荐
挽风82125 分钟前
Excel file format cannot be determined, you must specify an engine manually.
python
叫我:松哥1 小时前
基于网络爬虫的在线医疗咨询数据爬取与医疗服务分析系统,技术采用django+朴素贝叶斯算法+boostrap+echart可视化
人工智能·爬虫·python·算法·django·数据可视化·朴素贝叶斯
zylyehuo2 小时前
AirSim基础使用【Python】
python·drone
霜绛3 小时前
机器学习笔记(三)——决策树、随机森林
人工智能·笔记·学习·决策树·随机森林·机器学习
东方佑4 小时前
高效序列建模新突破:SamOut模型解读与21.79%损失改进
开发语言·python
ahauedu4 小时前
用Java 代码实现一个简单的负载均衡逻辑
java·python·负载均衡
工业甲酰苯胺4 小时前
Django集成Swagger全指南:两种实现方案详解
python·django·sqlite
豌豆花下猫4 小时前
Python 潮流周刊#112:欢迎 AI 时代的编程新人
后端·python·ai
您的通讯录好友5 小时前
TechGPT2部署
linux·人工智能·python·大模型·techgpt
z are7 小时前
PyTorch 模型开发全栈指南:从定义、修改到保存的完整闭环
人工智能·pytorch·python