Kom igångKom igång gratis

Ange parametrar och träna en modell

Prediktiva uppgifter delas in i två kategorier: regression och klassificering. I godisdatasetet är utfallet en kontinuerlig variabel som beskriver hur ofta ett godis valdes framför ett annat i en serie av 1-mot-1-matchningar. För att förutsäga detta värde (vinstprocenten) använder du en regressionsmodell.

I den här övningen anger du några parametrar med hjälp av en random forest-regressionsmodell, rfr.

Den här övningen är en del av kursen

Modellvalidering i Python

Visa kurs

Övningsinstruktioner

  • Lägg till en parameter i rfr så att antalet träd som byggs är 100 och det maximala djupet för dessa träd är 6.
  • Se till att modellen är reproducerbar genom att ange ett slumpmässigt tillstånd (random_state) på 1111.
  • Använd metoden .fit() för att träna random forest-regressionsmodellen med X_train som indata och y_train som svar.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Set the number of trees
rfr.____ = ____

# Add a maximum depth
rfr.____ = ____

# Set the random state
rfr.____ = ____

# Fit the model
rfr.____(____, ____)
Redigera och kör kod