Stel parameters in en train een model
Voorspellende taken vallen in twee categorieën: regressie of classificatie. In de candy-gegevensset is de uitkomst een continue variabele die beschrijft hoe vaak de snoepsoort werd gekozen boven een andere in een reeks 1-op-1 matchups. Om deze waarde (het winstpercentage) te voorspellen, gebruik je een regressiemodel.
In deze oefening specificeer je een paar parameters met behulp van een random forest regressiemodel rfr.
Deze oefening maakt deel uit van de cursus
Modelvalidatie in Python
Oefeninstructies
- Voeg parameters toe aan
rfrzodat het aantal te bouwen trees 100 is en de maximale diepte van deze trees 6. - Zorg dat het model reproduceerbaar is door een random state van
1111toe te voegen. - Gebruik de
.fit()-methode om het random forest regressiemodel te trainen metX_trainals invoergegevens eny_trainals response.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)