MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Bangun model_spec, sebuah spesifikasi pohon regresi.
  • Gunakan kerangka data chocolate_train, gunakan model_spec untuk melatih pohon regresi yang memprediksi final_grade hanya 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
Edit dan Jalankan Kode