Sprawdzanie dokładności predykcji
Czas sprawdzić, jak dobrze wytrenowany model radzi sobie z predykcją! Użyj zbioru testowego, aby ocenić dokładność drzewa decyzyjnego model za pomocą metody score().
To ćwiczenie jest częścią kursu
HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Instrukcje do ćwiczenia
- Zastosuj model drzewa decyzyjnego
model, aby dopasowaćfeaturesdotargetw zbiorze treningowym. - Sprawdź dokładność
score()predykcji dla zbioru treningowego. - Sprawdź dokładność
score()predykcji dla zbioru testowego.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Apply a decision tree model to fit features to the target in the training set
model.fit(____,____)
# Check the accuracy score of the prediction for the training set
model.____(features_train,____)*100
# Check the accuracy score of the prediction for the test set
model.____(____,____)*100