Ocena modelu
Pamiętasz dane testowe, które odłożyłeś(-aś) na bok jeszcze w rozdziale 3? Nareszcie nadszedł czas, żeby sprawdzić na nich swój model! Możesz użyć tego samego ewaluatora, którego użyłeś(-aś) do dopasowania modelu.
To ćwiczenie jest częścią kursu
Podstawy PySpark
Instrukcje do ćwiczenia
- Użyj modelu do wygenerowania prognoz, stosując
best_lr.transform()na danychtest. Zapisz wynik jakotest_results. - Wywołaj
evaluator.evaluate()natest_results, aby obliczyć wartość AUC. Wyświetl wynik.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Use the model to predict the test set
test_results = best_lr.____(____)
# Evaluate the predictions
print(evaluator.evaluate(____))