1. Learn
    2. /
  2. Cursuri
    3. /
  3. Machine Learning cu modele bazate pe arbori în Python
Connected

exercițiu

Utilizarea entropiei ca criteriu

În acest exercițiu, vei antrena un arbore de clasificare pe setul de date Wisconsin Breast Cancer, folosind entropia ca criteriu de informație. Vei utiliza toate cele 30 de caracteristici din set, care este împărțit în 80% antrenament și 20% testare.

X_train, precum și array-ul de etichete y_train sunt disponibile în spațiul tău de lucru.

Instrucțiuni

100 XP

  • Importă DecisionTreeClassifier din sklearn.tree.

  • Instanțiază un DecisionTreeClassifier numit dt_entropy cu o adâncime maximă de 8.

  • Setează criteriul de informație la 'entropy'.

  • Antrenează dt_entropy pe setul de antrenament.