Scikit-learn
- 모델 생성 : 사용하고자 하는 모델 import 후 아래아 같이 코딩 model = 모델명
- 학습 : model.fit(x, y)
- 예측 : prediction = model.predict(x2)
- 예제 :
from sklearn.linear_model import LinearRegression
model = LinearRegression()
model.fit(x, y)
prediction = model.predict(x2)
실습 :
import numpy as np
from sklearn.linear_model import LinearRegression
x = np.arange(10).reshape(-1, 1)
y = (2*x + 1).reshape(-1, 1)
# 모델 선언
model = LinearRegression()
# 학습
model.fit(x, y)
# 예측
prediction = model.predict([[10.0]])
prediction
''' 출력
array([[21.]])
'''