IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Aggiungi un parametro a rfr in 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 con X_train come dati in input e y_train come 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.____(____, ____)
Modifica ed esegui il codice