基于Pytorch实现逻辑回归_python_

import torch as t import matplotlib.pyplot as plt from torch import nn from torch.autograd import Variable import numpy as np     # 构造数据集 n_data = t.ones(100, 2) # normal（）返回一个张量，张量里面的随机数是从相互独立的正态分布中随机生成的。 x0 = t.normal(2*n_data, 1) y0 = t.zeros(100) x1 = t.normal(-2*n_data, 1) y1 = t.ones(100)   # 把数据给合并以下，并且数据的形式必须是下面形式 x = t.cat((x0, x1), 0).type(t.FloatTensor) y = t.cat((y0, y1), 0).type(t.FloatTensor)   # 观察制造的数据 plt.scatter(x.data.numpy()[:, 0], x.data.numpy()[:, 1], c=y.data.numpy(), s=100, lw=0) plt.show()   # 建立逻辑回归 class LogisticRegression(nn.Module):     def __init__(self):         super(LogisticRegression, self).__init__()         self.lr = nn.Linear(2, 1)         self.sm = nn.Sigmoid()     def forward(self, x):         x = self.lr(x)         x = self.sm(x)         return x # 实例化 logistic_model = LogisticRegression() # 看GPU是否可使用，如果可以使用GPU否则不使用 if t.cuda.is_available():     logistic_model.cuda() # 定义损失函数和优化函数 criterion = nn.BCELoss() optimizer = t.optim.SGD(logistic_model.parameters(), lr=1e-3, momentum=0.9) # 训练模型 for epoch in range(1000):     if t.cuda.is_available():         x_data = Variable(x).cuda()         y_data = Variable(y).cuda()     else:         x_data = Variable(x)         y_data = Variable(y)         out = logistic_model(x_data)         loss = criterion(out, y_data)         print_loss = loss.data.item()         # 以0.5为阈值进行分类         mask = out.ge(0.5).float()         # 计算正确预测样本的个数         correct = (mask==y_data).sum()         # 计算精度         acc = correct.item()/x_data.size(0)         optimizer.zero_grad()         loss.backward()         optimizer.step()         # 每个200个epoch打印一次当前的误差和精度         if(epoch+1)%200==0:             print('*'*10)             # 迭代次数             print('epoch{}'.format(epoch+1))             # 误差             print('loss is {:.4f}'.format((print_loss)))             # 精度             print('acc is {:.4f}'.format(acc)) if __name__=="__main__":     logistic_model.eval()     w0, w1 = logistic_model.lr.weight[0]     w0 = float(w0.item())     w1 = float(w1.item())     b = float(logistic_model.lr.bias.item())     plot_x = np.arange(-7, 7, 0.1)     plot_y = (-w0*plot_x-b)/w1     plt.scatter(x.data.numpy()[:, 0], x.data.numpy()[:, 1], c=y.data.numpy(), s=100, lw=0)     plt.plot(plot_x, plot_y)     plt.show()