[파이토치 스터디] DataLoader 사용
📚 미니배치와 DataLoader 사용하기 파이토치에서는 DataLoader를 사용해서 미니 배치 형태로 쉽게 처리할 수 있다. import torch import torch.nn as nn import torch.nn.functional as F from torch.utils.data import TensorDataset # 텐서데이터셋 from torch.utils.data import DataLoader # 데이터로더 x_train = torch.FloatTensor([[73, 80, 75], [93, 88, 93], [89, 91, 90], [96, 98, 100], [73, 66, 70]]) y_train = torch.FloatTensor([[152], [185], [180], [196], [..
2022. 3. 2.
[파이토치 스터디] 경사하강법 구현, Class 사용하기
📚 선형회귀로 경사하강법 살펴보기 • 경사하강법은 위와 같은 cost function을 미분해서 기울기(gradient)가 가장 심한 지점을 구해서, 비용함수의 최저 지점으로 이동하는 방향을 찾아내는 알고리즘이다. 📌 변수 선언 import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim x_train = torch.FloatTensor([[1], [2], [3], [4], [5], [6]]) y_train = torch.FloatTensor([[2], [4], [6], [7], [11], [14]]) 📌 가중치, 편향 초기화 #%% 가중치, 편향 초기화 W = torch.zeros(1, requ..
2022. 3. 1.