Written by LSM
on September 05, 2018

TIL:선형회귀

선형회귀

지도학습에서 사용되는 대표적인 방법
종속 변수 y와 독립 변수 x의 선형 상관 관계를 분석하는 것 y = β0 + β1x + ε
β0 과 β1은 추정하고자 하는 모델 계수
계수는 MSE를 최소화하는 계수를 구함

code

import numpy as np
import matplotlib.pyplot as plt
from scipy import stats

x = np.array([1.5, 2, 2.5, 3, 3.5, 4, 4.5, 5, 5.5, 6])
y = np.array([10.35, 12.3, 13, 14.0, 16, 17, 18.2, 20, 20.7, 22.5])

gradient, intercept, r_val, p_val, std_err = stats.linregress(x,y)

mn = np.min(x)
mx = np.max(x)

x1 = np.linspace(mn, mx, 500)
y1 = gradient * x1 + intercept

plt.plot(x, y, 'ob')
plt.plot(x1, y1, '-r')
plt.show()

← → Top