MulaiMulai sekarang secara gratis

Menggunakan entropy sebagai kriteria

Dalam latihan ini, Anda akan melatih pohon klasifikasi pada himpunan data Wisconsin Breast Cancer dengan menggunakan entropy sebagai kriteria informasi. Anda akan melakukannya dengan menggunakan semua 30 fitur dalam himpunan data, yang dibagi menjadi 80% latih dan 20% uji.

X_train serta array label y_train tersedia di workspace Anda.

Latihan ini adalah bagian dari kursus

Machine Learning dengan Model Berbasis Pohon di Python

Lihat Kursus

Petunjuk latihan

  • Impor DecisionTreeClassifier dari sklearn.tree.

  • Instansiasi DecisionTreeClassifier dt_entropy dengan kedalaman maksimum 8.

  • Tetapkan kriteria informasi ke 'entropy'.

  • Latih dt_entropy pada himpunan latih.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import DecisionTreeClassifier from sklearn.tree
from ____.____ import ____

# Instantiate dt_entropy, set 'entropy' as the information criterion
dt_entropy = ____(____=____, ____='____', random_state=1)

# Fit dt_entropy to the training set
____.____(____, ____)
Edit dan Jalankan Kode