Allena un modello semplice
Come hai stabilito, stai affrontando un problema di regressione. Ora sei pronto per costruire un modello da inviare. Ma invece di creare il modello più semplice di Regressione Lineare come nelle diapositive, costruiamo un modello Random Forest pronto all’uso.
Userai la classe RandomForestRegressor dalla libreria scikit-learn.
Il tuo obiettivo è addestrare un modello Random Forest con i parametri di default sulle caratteristiche "store" e "item".
Questo esercizio fa parte del corso
Vincere una competizione Kaggle con Python
Istruzioni dell'esercizio
- Leggi i dati di train usando
pandas. - Crea un oggetto Random Forest.
- Allena il modello Random Forest sulle caratteristiche "store" e "item" usando "sales" come target.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
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['____'])