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
Övningsinstruktioner
- Lägg till en parameter i
rfrså 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 medX_trainsom indata ochy_trainsom 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.____(____, ____)