import torch from torch.autograd import Variable import matplotlib.pyplot as plt x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1) y = x.pow(2)+0.2*torch.rand(x.size()) x, y = Variable(x), Variable(y) class Net(torch.nn.Module): def __init__(self, n_fatures, n_hidden, n_output): super(Net, self).__init__() self.hidden = torch.nn.Linear(n_fatures, n_hidden) self.predict = torch.nn.Linear(n_hidden, n_output) def forward(self, x): x = torch.relu(self.hidden(x)) x = self.predict(x) return x net = Net(1, 10, 1) plt.ion() plt.show() optimizer = torch.optim.SGD(net.parameters(), lr=0.01) loss_Func = torch.nn.MSELoss() for t in range(100): prediction = net(x) loss = loss_Func(prediction, y) optimizer.zero_grad() loss.backward() optimizer.step() if t%5 == 0: plt.cla() plt.scatter(x.data.numpy(),y.data.numpy()) plt.plot(x.data.numpy(), prediction.data.numpy(), 'r-', lw=5) plt.text(0.5, 0, 'loss =%.4f'%loss.data, fontdict={'color':'red', 'size':'20'} ) plt.pause(0.5) plt.ioff() plt.show()
pytorch(1) 预测曲线
最新推荐文章于 2024-05-06 15:25:00 发布