数据与标签定义
读取数据,看看GDP和HappinessScore的线性关系如何。
用sample将数据集拆分成训练集traindata和测试集testdata,比例为(0.8)8:2。指定输入特征GDP和输出特征Score。plt将其展示出来
训练线性回归模型
指定迭代次数和学习率,将训练集传入线性回归模型中,通过train函数得到theta值和损失值
损失值应该是不断减少的,将其打印出来看看损失值的变化
得到线性回归方程
以上是训练集,接下来做测试集。用linspace方法等距抽取100个样本。根据样本数据x_predictions经过线性回归模型得到预测值y_predictions。plt将这条线打印出来