numpy 간단한 선형회귀 예제

##선형회귀(MSE)
 
%matplotlib inline
import numpy as np
import matplotlib.pyplot as plt
 
 
x = np.array([1, 2, 3, 4])
y = np.array([-4, 2, 0.9, 2.1])
 
A = np.vstack([x, np.ones(len(y))]).T
w,b = lin.lstsq(A,y)[0]
 
print('model:',w,'* x +',b)
 
plt.plot(x,y, 'x', label='orignal_data', markersize=10)
plt.plot(x, w * x + b , 'r', label='linearRegression')
plt.legend()
plt.grid(True)
plt.show()

결과