[Pytorch+딥러닝] 3-3. nn.Module과 클래스로 구현하기
·
ML & DL/딥러닝 기초
이전 챕터까지는 가설, 비용 함수를 직접 정의해서 모델을 구현했었음이번에는 Pytorch에서 이미 구현되어 있는 함수들을 불러와서 더 쉽게 회귀 모델들을 구현해보자! 1. 단순 선형 회귀 구현아래 두 함수를 활용할 예정!`nn.Linear()` : 선형회귀 모델이 구현되어있는 함수`nn.functional.mse_loss()` : 평균제곱오차(MSE) 계산이 구현되어있는 함수# 사용 예시 미리보기import torch.nn as nnmodel = nn.Linear(input_dim, output_dim) # 선형회귀모델import torch.nn.functional as Fcost = F.mse_loss(prediction, y_train) # 평균제곱오차 데이터 준비, 가설 선언이전과 동일하게 정답 함..