Pada latihan ini, akan dilakukan klasifikasi data yang dimiliki dengan teknik Decision Tree menggunakan dataset alpha.
Dataset alpha terdiri dari 3 atribut yaitu umur, berat badan, dan tinggi badan. Terdapat 2 kelas target pada dataset ini. Data ini digunakan untuk masalah klasifikasi, di mana kita memprediksi gender berdasarkan atribut-atribut yang diberikan.
Tahapan yang ada pada latihan ini antara lain:
- Ubah dataset ke dalam dataframe.
- Hapus kolom 'Id' pada dataframe serta pisahkan antara atribut dan label.
- Bagi dataset menjadi data latih dan data uji.
- Buat dan latih model Decision Tree.
- Lakukan pengujian model dengan menggunakan data uji.
- Lakukan prediksi dengan model yang telah dilatih.
- Visualisasi model Decision Tree yang telah dilatih.
Berikut merupakan hasil visualisasi dari model decision tree yang telah kita gunakan:
