⚙️ Tech/ML

[ML] 사이킷런 지도 학습 클래스 정리

fiftyline 2025. 2. 12. 16:22

 

 

모델종류 (모듈명) 모델 클래스 (분류) 모델 클래스 (예측)
선형 모델 (linear_model) - LogisticRegression - LinearRegression
- Ridge
- Lasso
- ElasticNet
서포트 벡터 머신 (svm) - SVC
- NuSVC
- LinearSVC
- SVR
k-최근접 이웃 (neighbors) - KNeighborsClassifier - KNeighborsRegressor
나이브베이즈 (naive_bayes) - GaussianNB
- MultinomialNB
- BernoulliNB
 
결정 나무 (tree) - DecisionTreeClassifier - DecisionTreeRegressor
앙상블 (ensemble) - RandomForestClassifier
- GradientBoostingClassifier
- StackingClassifier
- RandomForestRegressor
- GradientBoostingRegressor
- StackingRegressor
신경망 (neural_network)

- MLPClassifier - MLPRegressor



 

 

 

참고 서적 : GIL's LAB, 「파이썬을 활용한 머신러닝 자동화 시스템 구축」, 위키북스(2022)