上传者: 38691703
|
上传时间: 2022-05-14 15:02:05
|
文件大小: 59KB
|
文件类型: PDF
本文用于利用Pytorch实现神经网络的分类!!!
1.训练神经网络分类模型
import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import torch.nn.functional as F
import torch.utils.data as Data
torch.manual_seed(1)#设置随机种子,使得每次生成的随机数是确定的
BATCH_SIZE = 5#设置batch size
#1.制作两类数据
n_data = torch.ones( 1000,2 )
x0 =