pytorch下对简单的数据进行分类(classification)

看了Movan大佬的文字教程让我对pytorch的基本使用有了一定的了解,下面简单介绍一下二分类用pytorch的基本实现！

希望详细的注释能够对像我一样刚入门的新手来说有点帮助！

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

n_data = torch.ones(100,2) #生成一个100行2列的全1矩阵
x0 = torch.normal(2*n_data,1)#利用100行两列的全1矩阵产生一个正态分布的矩阵均值和方差分别是(2*n_data,1)
y0 = torch.zeros(100)#给x0标定标签确定其分类0

x1 = torch.normal(-2*n_data,1) #利用同样的方法产生第二个数据类别
y1 = torch.ones(100)#但是x1数据类别的label就标定为1

x = torch.cat((x0,x1),0).type(torch.FloatTensor)#cat方法就是将两个数据样本聚合在一起(x0,x1),0这个属性就是第几个维度进行聚合
y = torch.cat((y0,y1),).type(torch.LongTensor)#y也是一样

x = Variable(x)#将它们装载到Variable的容器里
y = Variable(y)#将它们装载到Variable的容器里

#plt.scatter(x.data.numpy()[:,0],x.data.numpy()[:,1],c=y.data.numpy(),s=100,lw=0,cmap='RdYlGn')
#plt.show()

class Net(torch.nn.Module):#开始搭建一个神经网络
	def __init__(self,n_feature,n_hidden,n_output):#神经网络初始化，设置输入曾参数，隐藏曾参数，输出层参数
		super(Net,self).__init__()#用super函数调用父类的通用初始化函数初始一下
		self.hidden = torch.nn.Linear(n_feature,n_hidden)#设置隐藏层的输入输出参数，比如说输入是n_feature,输出是n_hidden
		self.out    = torch.nn.Linear(n_hidden,n_output)#同样设置输出层的输入输出参数

	def forward(self,x):#前向计算过程
		x = F.relu(self.hidden(x)) #样本数据经过隐藏层然后被Relu函数掰弯！
		x = self.out(x)经过输出层返回
		return x

net = Net(n_feature=2,n_hidden=10,n_output=2) #two classification has two n_features#实例化一个网络结构
print(net)

optimizer = torch.optim.SGD(net.parameters(),lr=0.02) #设置优化器参数,lr=0.002指的是学习率的大小
loss_func = torch.nn.CrossEntropyLoss()#损失函数设置为loss_function

plt.ion()

for t in range(100):
	out = net(x)#100次迭代输出
	loss = loss_func(out,y)#计算loss为out和y的差异

	optimizer.zero_grad()#清除一下上次梯度计算的数值
	loss.backward()#进行反向传播
	optimizer.step()#最优化迭代

	if t%2 == 0:
		plt.cla()
		prediction = torch.max(out,1)[1] ##返回每一行中最大值的那个元素，且返回其索引  torch.max()[1]， 只返回最大值的每个索引
		pred_y = prediction.data.numpy().squeeze()
		target_y = y.data.numpy()
		plt.scatter(x.data.numpy()[:,0],x.data.numpy()[:,1],c=pred_y,s=100,lw=0,cmap='RdYlGn')
		accuracy = float((pred_y == target_y).astype(int).sum())/float(target_y.size)
		plt.text(1.5,-4,'Accuracy=%.2f'%accuracy,fontdict={'size':20,'color':'red'})
		plt.pause(0.1)
plt.ioff()
plt.show()

最终运行出来的结果在下面：