机器学习-决策树(基于Python实现)

机器学习-决策树(基于Python实现)

概述

决策树(decision tree)是机器学习中一种非常重要的模型,主要的策略是分而治之。

决策树一般被用于分类问题(分类问题即对应离散的分布律,可以直观地求对应的其信息熵),但有时也可用于回归任务(如CART回归树)。

单变量决策树(univariate decision tree)的分类边界总是垂直于坐标轴的(如:敲声=清脆?, 密度<3.0560?),这样的分类边界可以容易地化为有限个,对于连续变量,将出现的值划分到不同bins即可。当边界形状复杂时,往往需要采用多变量决策树(multivariate decision tree),每次将采用特征的线性组合,而不是仅仅采用某一个最佳特征作为划分依据。

这里的理论和算法部分基于西瓜书第四章,实现采用的是python和常用的矩阵计算、数据分析库: numpy, pandas, matplotlib, pytorch等。

设计代码

https://gitee.com/yangtao2019yt/pytorch_learning/tree/master/machine_learning_by_torch/decision_tree

设计方案

ID3

ID3_tree_on_xigua2d0

C4.5

C4d5_tree_on_xigua2d0

CART

CART_tree_on_xigua2d0

其他

asd

参考资料

[1] 机器学习, 清华大学出版社, 周志华,

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
决策树是一种基本的分类和回归方法,它可以通过训练数据集来建立一个树形结构的模型,用于预测新数据的分类或数值。 在Python中,我们可以使用scikit-learn库来实现决策树。下面是一个简单的例子,展示如何使用决策树来进行分类。 首先,我们需要准备训练数据集和测试数据集。假设我们有一个包含三个特征的数据集,其中每个样本都属于两个类别之一: ```python from sklearn.datasets import load_iris from sklearn.model_selection import train_test_split from sklearn.tree import DecisionTreeClassifier # 加载Iris数据集 iris = load_iris() # 创建训练数据集和测试数据集 X_train, X_test, y_train, y_test = train_test_split( iris.data, iris.target, test_size=0.3, random_state=42) # 创建决策树分类器 clf = DecisionTreeClassifier() # 训练模型 clf.fit(X_train, y_train) # 在测试数据集上进行预测 y_pred = clf.predict(X_test) # 计算准确率 accuracy = clf.score(X_test, y_test) print("Accuracy:", accuracy) ``` 上面的代码中,我们首先使用`load_iris()`函数加载了Iris数据集,然后使用`train_test_split()`函数将数据集分成训练数据集和测试数据集。接着,我们创建了一个`DecisionTreeClassifier`对象作为决策树分类器,并使用训练数据集来训练模型。最后,我们在测试数据集上进行了预测,并计算了模型的准确率。 需要注意的是,在实际应用中,我们通常需要对训练数据集进行一些预处理工作,比如特征选择、特征缩放等。 决策树的优点在于它们易于理解和解释,可以处理离散和连续特征,适用于多类别问题,且能够自动进行特征选择。但是,决策树也有一些缺点,比如容易过拟合、对噪声敏感等。在实际应用中,我们通常需要结合其他方法来提高模型的性能。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值