Latih pohon regresi
Seperti yang sudah Anda ketahui, decision tree adalah alat yang berguna untuk masalah klasifikasi. Selain itu, Anda juga dapat menggunakannya untuk memodelkan masalah regresi. Perbedaannya secara struktural adalah akan ada nilai numerik (alih-alih kelas) pada node daun.
Dalam latihan ini, Anda akan menggunakan himpunan data cokelat untuk memasangkan sebuah pohon regresi. Ini sangat mirip dengan yang sudah Anda lakukan di Bab 1 dengan himpunan data diabetes.
Data pelatihan chocolate_train tersedia di ruang kerja Anda.
Latihan ini adalah bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di R
Petunjuk latihan
- Bangun
model_spec, sebuah spesifikasi pohon regresi. - Gunakan kerangka data
chocolate_train, gunakanmodel_specuntuk melatih pohon regresi yang memprediksifinal_gradehanya dengan prediktor numerik dalam data.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
library(tidymodels)
# Build the specification
model_spec <- decision_tree() %>%
set_mode(___) %>%
set_engine(___)
# Fit to the data
model_fit <- model_spec %>%
___(formula = ___,
data = ___)
model_fit