楼主: xzm1945
1505 2

回归树预测是不是不能输出模型? [推广有奖]

  • 0关注
  • 0粉丝

本科生

56%

还不是VIP/贵宾

-

威望
0
论坛币
1 个
通用积分
0
学术水平
0 点
热心指数
0 点
信用等级
0 点
经验
918 点
帖子
55
精华
0
在线时间
99 小时
注册时间
2009-5-6
最后登录
2024-10-30

楼主
xzm1945 发表于 2022-2-28 16:55:30 |AI写论文

+2 论坛币
k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群
赵安豆老师微信:zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

感谢您参与论坛问题回答

经管之家送您两个论坛币!

+2 论坛币
问题1请问,回归树进行预测,能输出树的模型么?



问题2
如果能输出树的构成的图和文本,但是客户想要一个能让他们带入样本数值验证计算的模型,是不是不可以呢,因为只有一个树的模型
二维码

扫码加我 拉你入群

请注明:姓名-公司-职位

以便审核进群资格,未注明则拒绝

关键词:样本数

沙发
正直者之死 发表于 2022-2-28 17:59:36
感觉都可以
  1. from sklearn.tree import DecisionTreeRegressor, export_graphviz
  2. import graphviz
  3. import joblib
  4. import numpy as np
  5. import os

  6. data_X = np.array([[np.random.random() for i in range(5)] for j in range(100)])
  7. data_y = np.array([np.random.random() for i in range(100)])

  8. DR = DecisionTreeRegressor()

  9. DR.fit(data_X, data_y)

  10. # 保存&读取
  11. joblib.dump(DR, 'tree.joblib')
  12. DR = joblib.load('tree.joblib')

  13. # 作图
  14. dot_data = export_graphviz(DR, out_file = 'tree.dot', feature_names = list('abcde'), filled = True, rounded = True, special_characters = True)

  15. graph = graphviz.Source(dot_data)

  16. os.system('dot -Tpng tree.dot -o tree.png')
  17. # os.system('dot -Tpng tree.dot -o tree.pdf')
复制代码

tree.png (1.29 MB)

tree.png

藤椅
xzm1945 发表于 2022-3-1 10:24:17
腻害 腻害!!!! 实在是高人,非常感谢!!!

您需要登录后才可以回帖 登录 | 我要注册

本版微信群
加好友,备注cda
拉您进交流群
GMT+8, 2025-12-22 06:27