선형회귀_nn.Module

import torch.nn as nn #model = nn.Linear(input_dim, output_dim) import torch.nn.functional as F #cost = F.mse_loss(prediction, y_train) import torch import torch.nn as nn import torch.nn.functional as F torch.manual_seed(1) # 데이터 x_train = torch.FloatTensor([[1], [2], [3]]) y_train = torch.FloatTensor([[2], [4], [6]]) # 모델을 선언 및 초기화. 단순 선형 회귀이므로 input_dim=1, output_dim=1. model = nn.Linear(1,1) print(list(model.parameters())) # optimizer 설정.

경사 하강법 SGD를 사용하고 learning rate를 의미하는 lr은 0.01 optimize...