Mengekspor pohon

Dalam tugas klasifikasi Decision Tree, overfitting biasanya terjadi akibat pohon yang tumbuh terlalu dalam. Seperti yang ditunjukkan oleh perbandingan skor akurasi pada himpunan latih dan uji, hasil Anda mengalami overfitting. Hal ini juga dapat dilihat dari visualisasi pohon.

Pada latihan ini, Anda akan mengekspor decision tree ke dalam dokumen teks, yang kemudian dapat digunakan untuk visualisasi.

Latihan ini adalah bagian dari kursus

Analitik SDM: Memprediksi Perputaran Karyawan dengan Python

Lihat Kursus

Petunjuk latihan

Impor fungsi export_graphviz() dari submodul sklearn.tree.
Latih model pada data pelatihan.
Ekspor visualisasi ke berkas tree.dot.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import the graphical visualization export function
from sklearn.____ import export_graphviz

# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)

# Export the tree to a dot file
____(model,"tree.____")

Edit dan Jalankan Kode