Vorhersagen erzeugen und RMSE berechnen
Jetzt, da wir ein Modell haben, das auf unseren Daten trainiert und per Cross-Validation getuned wurde, können wir sehen, wie es sich auf dem test-DataFrame schlägt. Dafür berechnen wir die RMSE.
Hinweis am Rande: Das Erzeugen der Testvorhersagen dauert bei diesem Datensatz mehr als nur ein paar Minuten. Aus diesem Grund wurden die Testvorhersagen bereits erzeugt und hier als DataFrame test_predictions bereitgestellt. Zur Orientierung: Sie werden mit diesem Code erzeugt: test_predictions = best_model.transform(test).
Diese Übung ist Teil des Kurses
<Kurs>Recommendation Engines mit PySpark erstellen</Kurs>Übungsanweisungen
- Der DataFrame
test_predictionsenthält Vorhersagen, die unser per Cross-Validation trainiertes ALS-Modell mit dem zuvor erstelltentest-Set generiert hat. Verwende die Methode.show(), um ihn dir anzusehen und zu prüfen, ob die Vorhersagen plausibel wirken. - Verwende den zuvor erstellten
evaluator, um dieRMSEzu berechnen, indem du die Methode.evaluate()auf den erzeugtentest_predictionsaufrufst. Nenne das ErgebnisRMSE. - Gib die
RMSEaus.
Interaktive praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# View the predictions
test_predictions.____()
# Calculate and print the RMSE of test_predictions
RMSE = evaluator.____(____)
print(____)