本文用于利用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 =
1