GridSearchCV

이재은·2024년 7월 26일

https://velog.io/@hyunicecream/GridSearchCV란-어떻게-사용할까

xgb = XGBClassifier()
param_xgb = {"max_depth": [10,30,50],
              "min_child_weight" : [1,3,6,10],
              "n_estimators": [200,300,500,1000]
              }    
              
gscv_xgb  = GridSearchCV (estimator = xgb, param_grid = param_xgb, scoring ='accuracy', cv = 3, refit=True, n_jobs=1, verbose=2)
  • 모델,
  • 파라미터값
  • cv : 교차검증 kfold수,
  • verbose : GridSearchCV의 iteration시마다 수행 결과 메시지를 출력
  • scoring : 찾으려는 값(accuracy..)

  • 모델(gscv_xgb).bestparams : 최적의 파라미터 값을 보여줍니다.
  • 모델.bestscore : 예측 정확도를 보여줍니다.
profile
Dare to be an optimist

0개의 댓글