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

Connected

exercițiu

Entropie vs. indicele Gini

În acest exercițiu vei compara acuratețea pe setul de testare a lui dt_entropy cu acuratețea unui alt arbore numit dt_gini. Arborele dt_gini a fost antrenat pe același set de date, cu aceiași parametri, cu excepția criteriului de informație, care a fost setat la indicele Gini folosind cuvântul cheie 'gini'.

X_test, y_test, dt_entropy, precum și accuracy_gini, care corespunde acurateței obținute de dt_gini pe setul de testare, sunt disponibile în spațiul tău de lucru.

Instrucțiuni

100 XP
  • Importă accuracy_score din sklearn.metrics.
  • Prezice etichetele setului de testare pentru dt_entropy și atribuie rezultatul variabilei y_pred.
  • Evaluează acuratețea pe setul de testare a lui dt_entropy și atribuie rezultatul variabilei accuracy_entropy.
  • Analizează valorile accuracy_entropy și accuracy_gini.