import torch import time x_data=torch.tensor([[1.0],[2.0],[3.0],[4.0],[5.0],[6.0]]) y_data=torch.tensor([[9600.0],[9550.0],[9382.0],[9250.0],[9200],[9150.0]]) class linearModel(torch.nn.Module): def __init__(self): super(linearModel,self).__init__() self.linear=torch.nn.Linear(1,1) def forward(self,x): return self.linear(x) model=linearModel() criterion=torch.nn.MSELoss(size_average=True) optim=torch.optim.SGD(model.parameters(),lr=0.01) if __name__ == '__main__': for epoch in range (200): y_pred=model(x_data) loss=criterion(y_pred,y_data) print(epoch,loss.item()) time.sleep(0.03) optim.zero_grad() loss.backward() optim.step() x=torch.tensor([[9.0]]) y=model(x) print(f"y={y}")
python预算房价
最新推荐文章于 2024-07-12 16:16:27 发布