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 merupakan bagian dari kursus
Memenangi Kompetisi Kaggle dengan Python
Instruksi 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 langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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['____'])