二分类问题,0表示没通过,1表示通过,用BCEloss作为损失
import torch
import matplotlib.pyplot as plt
import numpy as np
x_data = torch.Tensor([[1.0], [2.0], [3.0]])
y_data = torch.Tensor([[0], [0], [1]]) #0表示没通过,1表示通过了
class LogisticRegressionModel(torch.nn.Module):
def __init__(self):
super(LogisticRegressionModel, self).__init__()
self.linear = torch.nn.Linear(1,1)
def forward(self,x):
y_pred = torch.sigmoid(self.linear(x))
return y_pred
model = LogisticRegressionModel()
criterion = torch.nn.BCELoss(reduction='sum') #torch新版本
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)
for epoch in range(1000):
y_pred = model(x_data) #forward:predict
loss = criterion(y_pred,y_data) #forward:loss
print(epoch,loss.item())
optimizer.zero_grad()
loss.backward() # backward:自动计算梯度
optimizer.step() #update 参数,更新w和b的值
print('w = ', model.linear.weight.item())
print('b = ', model.linear.bias.item())
x_test = torch.Tensor([[4.0]])
y_test = model(x_test)
print('y_pred=', y_test.data)
x = np.linspace(0,10,200) #(start,stop,num)
x_t = torch.Tensor(x).view((200,1)) #类似于numpy的reshape,变成200行*1列的矩阵
y_t = model(x_t)
y = y_t.data.numpy() #tensor.numpy将Tensor转化为ndarray,.item()将一个Tensor变量转化为python标量(int float等)
plt.plot(x,y)
plt.plot([0,10],[0.5,0.5],c='r')
plt.xlabel('Hours')
plt.ylabel('Probability of Pass')
plt.grid()
plt.show()