
인공지능 신경망
·
KT AIVLE School/언어지능 딥러닝
1. 선형회귀import torchx_train = torch.FloatTensor([[1,1],[2,2],[3,3]])y_train = torch.FloatTensor([[10],[20],[30]])W = torch.randn([2,1], requires_grad=True)b = torch.randn(1, requires_grad=True)optimizer = torch.optim.SGD([W,b], lr=0.01)# [딥러닝 1단계] 모델을 만든다. Model Setupdef H(x): model = torch.matmul(x,W) +b return model# [딥러닝 2단계] 학습을 시킨다. Training - W,b찾기for i in range(2000): cost = torch.mean(..