那么线性回归中最难的部分也就是模型训练的部分——怎么寻找到最适合的斜率和截距,也就是公式中的
线性回归实现(不调用sklearn库)
首先设定数据,是员工的工龄(年限)对应薪水(千元)的数据,使用散点图观察一下大致是否符合线性回归的情况...2
print('{:4}, w0:{:.8f}, w1:{:.8f}, loss:{:.8f}'.format(i+1,w0,w1,loss))
# 计算w0和W1方向上的偏导数...mode1.fit(输入,输出) # 通过梯度下降法计算模型参数
# 预测输出
# 输入array是一个二维数组,每一行是一个样本,每一列是一个特征。...在预测时,要传入一个二维数组,也就是要预测的样本,系统会计算后输出。...Samples')
plt.plot(x,pred_train_y,color='orangered',label='Regression Line')
plt.legend()
输出结果如下图,可以看出拟合的回归线与我们上面手动编写的线性回归模型效果相同