Atur parameter dan latih model
Tugas prediktif terbagi dalam dua kategori: regresi atau klasifikasi. Pada himpunan data permen, keluaran adalah variabel kontinu yang menggambarkan seberapa sering suatu permen dipilih dibanding permen lain dalam serangkaian pertandingan 1-lawan-1. Untuk memprediksi nilai ini (persentase kemenangan), Anda akan menggunakan model regresi.
Dalam latihan ini, Anda akan menentukan beberapa parameter menggunakan model random forest regression rfr.
Latihan ini adalah bagian dari kursus
Validasi Model di Python
Petunjuk latihan
- Tambahkan parameter pada
rfrsehingga jumlah pohon yang dibangun adalah 100 dan kedalaman maksimum pohon tersebut adalah 6. - Pastikan model dapat direproduksi dengan menambahkan random state
1111. - Gunakan metode
.fit()untuk melatih model random forest regression denganX_trainsebagai data masukan dany_trainsebagai respons.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)