기본 교차 성능 검증(RandomForestClassifier)

권형준·2026년 1월 30일

머신러닝

목록 보기
18/18

StratifiedKFold를 사용하여 훈련 데이터를 4개의 폴드로 나누고, 각 폴드에서의 정확도, 정밀도, 재현율, F1 점수를 계산

  1. features_org 로 train 칼럼중 Outcome 칼럼을 제외한 다른 칼럼 선택

  2. train_x 데이터는 Outcome 칼럼 제외 데이터, test_y데이터는 Outcome 칼럼 데이터

  3. StratifiedKFold로 4개의 데이터로 자르는 모델 생성

  4. RandomForest 분류 모델(classifier)로 맞는지 아닌지 확인. cross_validate 함수 사용

profile
천리길도 한발짝부터

0개의 댓글