Pytorch nn.Module로 단순 선형회귀 구현하기 nn.Linear()


Pytorch nn.Module로 단순 선형회귀 구현하기 nn.Linear()

이전 포스팅에서 말씀드렸던대로, pytorch의 nn.Module을 이용하여 선형회귀를 구현해보겠습니다. hypothesis, cost를 직접 정의해서 기초부터 모델을 구현하는 방법도 있지만, 본 포스팅에서는 파이토치(nn.Module)에서 이미 구현되어 제공되는 함수들을 불러와 선형회귀 모델을 구현해보려고 합니다. 기초부터 모델을 쌓는 방법을 공부해보고 싶으시다면 하단 링크를 참조하시면 될 것 같습니다. 오늘은 간단하게 단순 선형회귀를 구현해보고, 다음 포스팅에서 다중 선형회귀를 구현해보겠습니다. ** 본 포스팅은 pc 버전에 최적화 되어 있습니다. ** 단순 선형회귀 구현 nn.Linear() ## 기본 셋팅 import torch import torch.nn as nn import torch.nn.functional as F torch.manual_seed(1) 선형회귀 모델을 구현하는데 필요한 라이브러리를 불러와줍니다. torch.manual_seed()는 코드를 여러번 돌려...


#linear #딥러닝입문 #딥러닝 #대학생 #단순선형회귀 #공부블로그 #regression #pytorch #nn #통계학과

원문링크 : Pytorch nn.Module로 단순 선형회귀 구현하기 nn.Linear()