Data Science/python & pytorch

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

sunnyshiny 2023. 3. 6. 15:52
728x90

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

728x90