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