Latih model sederhana
Seperti yang telah Anda tentukan, Anda sedang menghadapi masalah regresi. Jadi, sekarang Anda siap membangun model untuk pengumpulan berikutnya. Namun kali ini, alih-alih membangun model Linear Regression paling sederhana seperti pada slide, mari kita bangun model Random Forest siap pakai.
Anda akan menggunakan kelas RandomForestRegressor dari pustaka scikit-learn.
Tujuan Anda adalah melatih model Random Forest dengan parameter default pada fitur "store" dan "item".
Latihan ini adalah bagian dari kursus
Memenangi Kompetisi Kaggle dengan Python
Petunjuk latihan
- Baca data latih menggunakan
pandas. - Buat objek Random Forest.
- Latih model Random Forest pada fitur "store" dan "item" dengan "sales" sebagai target.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
import pandas as pd
from sklearn.ensemble import RandomForestRegressor
# Read the train data
train = ____.____('train.csv')
# Create a Random Forest object
rf = ____()
# Train a model
rf.fit(X=train[['store', ____]], y=train['____'])