[sklearn] sklearn model의 일반적인 형태와 모델들

Estimator [학습: fit, 예측: predict]

• Classifier
  • DecisionTreeClassifier
  • RandomForestClassifier
  • GradientBoostingClassifier
  • GaussianNB
  • SVC
• Regressor
  • LinearRegression
  • Ridge
  • Lasso
  • RandomForestRegressor
  • GradientBosstingRegressor

from sklearn.module import Model
model = Model()
model.fit(X, y)
prediction = model.predict(x_new)

Model에 사용하고자 하는 sklearn model명을 넣고 train data를 fit으로 학습시켜주고, 그후 test or validation data를 예측해주는 것이 sklearn에서 일반적으로 사용되는 형태이다.

 

Reference 

파이썬 머신러닝 완벽 가이드

sklearn document