LoslegenKostenlos starten

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>
Kurs ansehen

Übungsanweisungen

  • Der DataFrame test_predictions enthält Vorhersagen, die unser per Cross-Validation trainiertes ALS-Modell mit dem zuvor erstellten test-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 die RMSE zu berechnen, indem du die Methode .evaluate() auf den erzeugten test_predictions aufrufst. Nenne das Ergebnis RMSE.
  • Gib die RMSE aus.

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(____)
Code bearbeiten und ausführen