Setează parametrii și antrenează un model
Sarcinile predictive se împart în două categorii: regresie sau clasificare. În setul de date despre bomboane, rezultatul este o variabilă continuă care descrie cât de des a fost aleasă o bomboană în fața alteia într-o serie de dueluri 1-la-1. Pentru a prezice această valoare (procentul de victorii), vei folosi un model de regresie.
În acest exercițiu, vei specifica câțiva parametri folosind un model de regresie cu păduri aleatoare rfr.
Acest exercițiu face parte din cursul
Validarea modelelor în Python
Instrucțiuni pentru exercițiu
- Adaugă un parametru la
rfrastfel încât numărul de arbori construiți să fie 100, iar adâncimea maximă a acestora să fie 6. - Asigură-te că modelul este reproductibil adăugând o stare aleatoare cu valoarea
1111. - Folosește metoda
.fit()pentru a antrena modelul de regresie cu păduri aleatoare, cuX_trainca date de intrare șiy_trainca răspuns.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)