前言
用pytorch,实现简单的线性回归。
# 一、用torch中内置的方法进行:
- 用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.进行训练和训练结果的展示: