Imposta i parametri e addestra un modello
I compiti predittivi rientrano in due categorie: regressione o classificazione. Nel dataset delle caramelle, l’outcome è una variabile continua che descrive la frequenza con cui una caramella è stata scelta rispetto a un’altra in una serie di sfide 1-contro-1. Per prevedere questo valore (la percentuale di vittorie), userai un modello di regressione.
In questo esercizio, specificherai alcuni parametri usando un modello di random forest per la regressione rfr.
Questo esercizio fa parte del corso
Validazione dei modelli in Python
Istruzioni dell'esercizio
- Aggiungi un parametro a
rfrin modo che il numero di alberi costruiti sia 100 e la profondità massima di questi alberi sia 6. - Assicurati che il modello sia riproducibile impostando un random state pari a
1111. - Usa il metodo
.fit()per addestrare il modello di random forest per la regressione conX_traincome dati in input ey_traincome vettore di risposta.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)