养生 装修 购物 美食 感冒 便秘 营销 加盟 小吃 火锅 管理 创业 搭配 减肥 培训 旅游

机器学习——用回归树 拟合正弦曲线

时间:2024-11-11 21:10:22

、在二维平面上来观察决策树是怎样拟合一条曲线的。我们用回归树来拟合正弦曲线,并添加一论蜈奈糍些噪声来 观察回归树的表现

工具/原料

开发工具Jupyter

方法/步骤

1、首先运行魔法命令,准备展示开发环境和第三包信息。如图示:

机器学习——用回归树 拟合正弦曲线

2、输入命令打印相关信息。如图示:

机器学习——用回归树 拟合正弦曲线

3、执行命令后。打印信息如图示:

机器学习——用回归树 拟合正弦曲线

4、开始进入正题。

5、导入相关模块,如图示:

机器学习——用回归树 拟合正弦曲线

6、用numpy模块生成正弦函数的训练数据。

7、方法如图示:

机器学习——用回归树 拟合正弦曲线

8、训练集数据生成好后,接下来生成回归树模型所需的测试集数据。

9、测试集生成如图示:

机器学习——用回归树 拟合正弦曲线

10、生成后,可以查看生成数据的相关信息.

11、正弦函数横坐标X的数据形状,如图示:

机器学习——用回归树 拟合正弦曲线

12、正弦函数横坐标X的部分数据,如图示:

机器学习——用回归树 拟合正弦曲线

13、正弦函数y的数据形状,如图示:

机器学习——用回归树 拟合正弦曲线

14、正弦函数y的部分数据,如图示:

机器学习——用回归树 拟合正弦曲线

15、正弦函数测试集数据的形状如图示:

机器学习——用回归树 拟合正弦曲线

16、测试集的部分数据如图示:

机器学习——用回归树 拟合正弦曲线

17、数据准备好后,进行下一步。

18、创建回归数模型。

19、由于需要两个模型做对比,因此实例化两个回归树实例。如图示:

机器学习——用回归树 拟合正弦曲线

20、接下来导入训练集数据,训练回归树模型。如图示:

机器学习——用回归树 拟合正弦曲线

21、模型训练完毕后,导入测试集数据,获取测试结果。如图示:

机器学习——用回归树 拟合正弦曲线

22、采集测试集在模型下的分类结果的得到是一组数据,为了方便观察,用matplotlib绘制为图像。

23、绘制图像代码如图示:

机器学习——用回归树 拟合正弦曲线

24、回归树拟合正弦曲的结果如图示:

机器学习——用回归树 拟合正弦曲线

25、从图中可以看出回归树的最大深度为2时,更加拟合正弦曲线。

© 一点知识