使用python实现多项式回归,没有使用sklearn等机器学习框架,目的是帮助理解算法的原理。
使用一个简单的数据集来模拟,只有几条数据。
代码
- 从数据集中读取X和y。
- 为X添加二次方项,用Z替换。
- 给Z添加 1 列,初始化为 1 ,用来求偏置项。
- 划分训练集和测试集。
- 将Z和y的训练集转换为矩阵形式。
- 和线性回归类似,使用正规方程法,先验证矩阵的可逆性。
- 去掉Z中全为1的列。
- 使用测试集验证模型。
- Z和y的训练集转换回数组形式。
- 打印结果和图片。
运行结果
从图中看出数据分布在一条抛物线附近。
最终得到的模型为:
如果需要代码和数据集,请扫描下面二维码关注公众号【AI developer】,回复【代码】即可免费获取。
原文地址:https://www.cnblogs.com/ai-developer/p/10503384.html
时间: 2024-10-11 01:06:56