back to basics - simple linear regression

be1le·2022년 4월 22일
4

ML

목록 보기
7/22
post-thumbnail

이번 시간에는 단순 선형회귀에 대한 실습을 구현해볼 것이다!

Importing the libraries


np,plt,pd는 꿈에서도 import 해오는것 같다.

Importing the dataset


연차에 따른 연봉에대한 데이터셋을 import해왔다. 늘 그렇듯 x,y에 할당해 주면된다.!

Splitting the dataset into the Training set and Test set

X,y만 지정해주고 test size 는 1/3으로 설정 하였다.

Training the Simple Linear Regression model on the Training set


linear_modle 에서 LinearRegression을 import해왔다.
fit으로 입혀주기만 하면 모델이 학습된다..!

Predicting the Test set results


.predict로 예측값에 접근해보면

Visualising the Training set results


그래프 이름을 정해주고 x-label, y-label을 지정해주면
이런식으로 그래프를 그려줄 수 있다.

Visualising the Test set results


이렇게 결과까지 깔끔하게 확인할 수 있다. 가장 기본이지만,동시에 Reference를 보지않고 구현해보면서 좀더 회귀에대한 깊은 이해를 얻은것 같은 기분이 든다! :)

profile
그저 그런 개발자가 되지 않겠습니다.

0개의 댓글