Pytorch入门实例

数据集是受教育年限和收入,如下图

代码如下

python 复制代码
import torch
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import torch.nn as nn
import torch.optim as optim

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

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

learning_rate = 0.0001
model = nn.Linear(1,1)
loss_fn = nn.MSELoss()
optimizer = optim.SGD(model.parameters(),lr=learning_rate)

for epoch in range(50):
    for x,y in zip(X,Y):
        output = model(x)
        loss = loss_fn(output,y)
        loss.backward()
        optimizer.step()
        optimizer.zero_grad()

plt.scatter(data.Education,data.Income)
plt.plot(X.numpy(),model(X).detach().numpy(),c='r')
plt.xlabel('Education')
plt.ylabel('Income')
plt.show()

输出如下

相关推荐
IT_陈寒7 分钟前
Python+AI实战:用LangChain构建智能问答系统的5个核心技巧
前端·人工智能·后端
亚马逊云开发者24 分钟前
Amazon Bedrock AgentCore Memory:亚马逊云科技的托管记忆解决方案
人工智能
扶尔魔ocy26 分钟前
python程序打包成win的exe应用(以OCR应用为例)
python·ocr·中文识别
言之。28 分钟前
Chroma 开源的 AI 应用搜索与检索数据库(即向量数据库)
数据库·人工智能·开源
Psycho_MrZhang35 分钟前
自定义层和读写文件
pytorch·python·深度学习
tomlone36 分钟前
《AI的未来:从“召唤幽灵”到学会反思》
人工智能
编码浪子39 分钟前
对LlamaFactory的一点见解
人工智能·机器学习·数据挖掘
长桥夜波41 分钟前
【第十八周】机器学习笔记07
人工智能·笔记·机器学习
似水流年 光阴已逝1 小时前
从Excel姓名匹配案例学Python:由点及面的系统化学习指南
开发语言·python·excel
这里有鱼汤1 小时前
量化的困局:当所有人都在跑同一个因子时,我们还能赚谁的钱?
后端·python