Acum vei încerca să construiești un model bazat pe arbori de decizie. Arborele de decizie reprezintă un set de reguli if-else învățate automat, care determină – în cazul predicției abandonului clienților din telecomunicații – dacă un client va renunța sau nu la servicii. Mai jos este un exemplu de arbore de decizie construit pe celebrul set de date Titanic privind supraviețuitorii.

Variabilele train_X, test_X, train_Y, test_Y din exercițiul anterior au fost deja încărcate. De asemenea, modulul tree și funcția accuracy_score au fost importate din biblioteca sklearn. Acum vei construi modelul și vei evalua performanța acestuia pe date nevăzute.

De ce să folosești ML în marketing? Strategii și cazuri de utilizare

Create Your Free Account