pytorch的碎碎念——分类问题

最新推荐文章于 2023-01-06 16:09:10 发布

大雾散去

最新推荐文章于 2023-01-06 16:09:10 发布

阅读量191

点赞数 1

文章标签： python 深度学习神经网络

本文链接：https://blog.csdn.net/qq_38880303/article/details/105450408

版权

本文基于莫烦Python的视频教程，介绍了如何使用PyTorch进行分类问题的处理，包括one-hot编码的概念和实现。在实践中遇到版本问题，通过调整代码解决了报错，并讨论了数据生成与模型训练，最终达到了高准确率的分类效果。

摘要由CSDN通过智能技术生成

根据b站莫烦python的视频教学



import torch
import numpy as np
import matplotlib.pyplot as plt
from torch.autograd import Variable
import torch.nn.functional as F

n_data = torch.ones(100,2)#data的基数
#x0=torch.normal(2*n_data,int(1))
#x1=torch.normal(-2*n_data,int(1))##normal要将数变成tensor形式
x0=torch.normal(2*n_data,1)#数据的横坐标，纵坐标都包含在x0中
x1=torch.normal(-2*n_data,1)
y0=torch.zeros(100)#x0标签为0
y1=torch.ones(100)#x1标签为1
x=torch.cat((x0,x1),0).type(torch.FloatT