<자격증>/[빅데이터분석기사]
빅분기 작업형 2 모델링 - 회귀모델
9566
2021. 12. 3. 16:11
728x90
# 9. 회귀 모델
# 9-1. 선형 회귀(릿지/라쏘)
from sklearn.linear_model import LinearRegression
from sklearn.linear_model import Ridge
from sklearn.linear_model import Lasso
from sklearn.linear_model import ElasticNet
model = ElasticNet()
# 9-2. 결정나무
from sklearn.tree import DecisionTreeRegressor
model = DecisionTreeRegressor(random_state=0)
# 9-3. SVM
from sklearn.svm import SVR
model = SVR()
# 9-4. 랜덤포레스트
from sklearn.ensemble import RandomForestRegressor
model = RandomForestRegressor(max_depth=10, n_estimators=100, random_state=0)
# 9-5. Ada 부스팅
from sklearn.ensemble import AdaBoostRegressor
model = AdaBoostRegressor(random_state=0)
# 9-6. XG 부스팅
from xgboost import XGBRegressor
model = XGBRegressor(random_state=0)
728x90