Generowanie predykcji i obliczanie RMSE
Mamy już model wytrenowany na naszych danych i dostrojony za pomocą walidacji krzyżowej – czas sprawdzić, jak radzi sobie na ramce danych test. W tym celu obliczymy RMSE.
Warto wiedzieć, że generowanie predykcji testowych na tym zbiorze danych zajmuje kilka minut. Dlatego zostały one już wygenerowane i są dostępne jako ramka danych test_predictions. Dla przypomnienia – do ich wygenerowania użyto następującego kodu: test_predictions = best_model.transform(test).
To ćwiczenie jest częścią kursu
Budowanie silników rekomendacji w PySpark
Instrukcje do ćwiczenia
- Ramka danych
test_predictionszawiera predykcje wygenerowane przez nasz model ALS z walidacją krzyżową na podstawie zbiorutest, który utworzyliśmy wcześniej. Użyj metody.show(), aby przyjrzeć się jej zawartości i sprawdzić, czy predykcje wyglądają wiarygodnie. - Użyj
evaluator, który zbudowałeś wcześniej, aby obliczyćRMSE– wywołaj metodę.evaluate()na ramcetest_predictions. Zapisz wynik jakoRMSE. - Wyświetl wartość
RMSE.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# View the predictions
test_predictions.____()
# Calculate and print the RMSE of test_predictions
RMSE = evaluator.____(____)
print(____)