Välj den bästa modellen
I den här övningen ska du jämföra olika klassificerare och välja den som presterar bäst.
Datamängden – redan inläst och uppdelad i tränings- och testset – består av Pokémon med deras statistik, typer och huruvida de är legendariska eller inte. Målet för våra klassificerare är att förutsäga variabeln 'Legendary'.
Tre individuella klassificerare har anpassats till träningsdata:
clf_lrär en logistisk regression.clf_dtär ett beslutsträd.clf_knnär en 5-närmaste-grannar-klassificerare.
Eftersom klasserna är obalanserade – bara 65 av de 800 Pokémon i datamängden är legendariska – använder vi F1-poäng för att utvärdera prestandan. Scikit-learns f1_score() har importerats åt dig.
Den här övningen är en del av kursen
Ensemblemetoder i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Predict the labels of the test set
pred_lr = ____
pred_dt = ____
pred_knn = ____