PyTorch : 선형 회귀(Linear Regression)

1.import packages import numpy as np import torch import torch.nn as nn import torch.optim as optim import torch.nn.init as init from torch.autograd import Variable from visdom import Visdom viz = Visdom() 2.데이터 생성 num_data = 1000 num_epoch = 1000 #정규분포에서 가져온다 noise = init.normal(torch.FloatTensor(num_data,1),std=1) #균일분포에서 가져온다 x = init.uniform(torch.Tensor(num_data,1),-10,10) y = 2*x+3 y_noise = 2*x+3+noise 3.시각화 input_data = torch.cat([x,y_noise],1) win=viz.scatter( X = input_data, opts=dict(...