pytorch:分类

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

n_data = torch.ones(100,2)
x0 = torch.normal(2*n_data,1)#第一类数据  class0 x data (tensor),shape=(100,2)
y0 = torch.zeros(100)#第一类的标签         class0 y data (tensor),shape=(100,1)
x1 = torch.normal(-2*n_data,1)#第二类数据 class1 x data (tensor),shape=(100,1)
y1 = torch.ones(100)#第二类的标签          class1 y data (tensor),shape=(100,1)

#把x合在一起，当作数据
x = torch.cat((x0,x1),0).type(torch.FloatTensor) #FloatTensor = 32bit floating
#把y合在一起，当作标签
y = torch.cat((y0,y1),).type(torch.LongTensor)  #LongTensor = 64bit integer
#把x,y放在Variable里进行学习
x,y = Variable(x),Variable(y)

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

class Net(torch.nn.Module):
    def __init__(self,n_feature,n_hidden,n_output):
        super(Net,self).__init__()#此为官方步骤
        self.hidden = torch.nn.Linear(n_feature,n_hidden)
        self.predict = torch.nn.Linear(n_hidden,n_output)
        pass
    def forward(self,x):#前向传播
        x = F.relu(self.hidden(x))
        x = self.predict(x)
        return x


net = Net(2,10,2)#输入2个特征，隐藏层神经元10个，输出2个
print(net)

plt.ion()
plt.show()

optimizer = torch.optim.SGD(net.parameters(),lr=0.02)#优化参数,SGD为优化器
loss_func = torch.nn.CrossEntropyLoss()#这个损失函数处理分类问题

for t in range(200):  # 训练100次
    out = net(x)

    loss = loss_func(out, y)  # 预测值与真实值的误差，预测值一定写在前面

    optimizer.zero_grad()  # 求导前先将导数归零,即清除上一次的grad
    loss.backward()
    optimizer.step()

    if t % 2 == 0:
        plt.cla()
        prediction = torch.max(out, 1)[1]
        pred_y = prediction.data.numpy()
        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()

视频传送门
结果图：