1. Lära sig
  2. /
  3. Courses
  4. /
  5. Python으로 배우는 트리 기반 Machine Learning

Connected

exercise

RF 회귀 모델 학습하기

다음 연습에서는 Kaggle에서 제공하는 Bike Sharing Demand 데이터셋의 과거 기상 데이터를 사용해 워싱턴 D.C.의 Capital Bikeshare 프로그램 자전거 대여 수요를 예측해 보겠습니다. 이를 위해 random forests 알고리즘을 사용할 거예요. 첫 단계로, random forests 회귀 모델을 정의하고 훈련 세트에 적합해 봅니다.

데이터셋은 이미 전처리되어 80%는 훈련, 20%는 테스트로 분할되어 있어요. 특성 행렬 X_train과 타깃 배열 y_train이 작업 공간에 준비되어 있습니다.

Instruktioner

100 XP
  • sklearn.ensemble에서 RandomForestRegressor를 임포트하세요.

  • 25개의 트리로 구성된 RandomForestRegressor를 rf라는 이름으로 인스턴스화하세요.

  • 훈련 세트에 rf를 적합(fit)하세요.