8、决策树模型的超参数、构建、解释与可视化

决策树模型的超参数、构建、解释与可视化

1. 决策树模型的超参数

决策树模型中有许多超参数,部分用于控制模型过拟合,部分用于提高模型准确性。重要超参数可参考表 1。

参数作用 描述
控制过拟合 部分超参数可限制决策树的生长,避免模型对训练数据过度拟合
提高准确性 部分超参数可调整决策树的结构,使模型更准确地预测数据

2. 创建决策树模型

接下来创建两个决策树模型 dt1 dt2 ,并验证训练集和测试集的准确率。
- dt1 :无最大深度限制,可能会过拟合。
- dt2 :最大深度限制为 3,可避免过拟合。

import sklearn.tree as tree

# 创建 dt1 模型
dt1 = tree.DecisionTreeClassifier()
dt1.fit(xtrain, ytrain)
print(dt1.score(xtrain, ytrain))
print(dt1.score(xtest, ytest))

# 创建 dt2 模型
dt2 = tree.Decisio
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值