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

Connected

exercițiu

Evaluează arborele de clasificare

Acum că ai antrenat primul tău arbore de clasificare, e momentul să îi evaluezi performanța pe setul de testare. Vei face acest lucru folosind metrica acuratețe, care reprezintă proporția predicțiilor corecte obținute pe setul de testare.

Modelul antrenat dt din exercițiul anterior este încărcat în spațiul tău de lucru, împreună cu matricea de caracteristici X_test și vectorul de etichete y_test.

Instrucțiuni

100 XP
  • Importă funcția accuracy_score din sklearn.metrics.

  • Prezice etichetele setului de testare și atribuie vectorul obținut variabilei y_pred.

  • Calculează scorul de acuratețe al modelului dt pe setul de testare apelând accuracy_score() și atribuie valoarea variabilei acc.