Pytorch---使用Pytorch实现线性回归模型一、代码中的数据集可以点击以下链接进行下载二、代码运行环境三、数据集处理的代码如下所示四、模型的构建代码如下所示五、模型的训练代码如下所示六、模型的预测代码如下所示七、代码的运行结果如下所示
一、代码中的数据集可以点击以下链接进行下载百度网盘提取码:lala二、代码运行环境Pytorch-gpu==1.7.1Python==3.7三、数据集处理的代码如下所示import pandas as pdimport torch# 数据的读取def make_data(): data = pd.read_csv(r'dataset\dataset.csv') X = data.Education.values.reshape(-1, 1) Y = data.I