前言

用pytorch,实现简单的线性回归。

1. 用pandas读取事先准备好的csv文件，这个csv文件主要包含以下内容：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import torch

data = pd.read_csv('../Income1.csv')

对数据进行可视化的操作

2.对数据的类型进行转换，使其转换为tensor（这边我用了两种转换方式，可以选其一）

X = torch.tensor(data.Education.values.reshape(-1,1).astype(np.float32))
Y = torch.from_numpy(data.Income.values.reshape(-1,1).astype(np.float32))

3.用torch中内置的模型，损失函数，优化算法进行操作，如图所示：

4.进行训练和训练结果的展示：

2.分解写法