看了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()
最终运行出来的结果在下面:
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch下对简单的数据进行分类(classification) - Python技术站