RMSE-Evaluator erstellen
Jetzt, da du weißt, wie man ein Modell auf Trainingsdaten fitten und Testvorhersagen generieren kann, brauchst du eine Möglichkeit, die Modellgüte zu bewerten. Dafür bauen wir einen evaluator. Evaluatoren in Spark können auf verschiedene Arten erstellt werden. Für unsere Zwecke wollen wir einen regressionEvaluator, der die RMSE berechnet. Nachdem wir unseren regressionEvaluator gebaut haben, können wir das Modell auf unsere Daten fitten und Vorhersagen erzeugen.
Diese Übung ist Teil des Kurses
<Kurs>Recommendation Engines mit PySpark erstellen</Kurs>Übungsanweisungen
- Importiere das benötigte Paket
RegressionEvaluatoraus der Klassepyspark.ml.evaluation. - Vervollständige den
evaluator-Code und setze denmetric nameauf"rmse". SetzelabelColauf den Namen der Spalte in unserenratings-Daten, die die Bewertungen enthält (verwenderatings.columns, um die Spaltennamen zu sehen), und setze den Namen derprediction-Spalte auf"prediction". - Bestätige, dass der
evaluatorkorrekt erstellt wurde, indem du die drei Parameter daraus ausliest. Führe dazu die folgenden drei Codezeilen jeweils in einem print-Statement aus:evaluator.getMetricName()evaluator.getLabelCol()evaluator.getPredictionCol()
Interaktive praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# Import RegressionEvaluator
from pyspark.ml.evaluation import ____
# Complete the evaluator code
evaluator = RegressionEvaluator(metricName="____", labelCol="____", predictionCol="____")
# Extract the 3 parameters
print(evaluator.get____())
print(evaluator.get____())
print(evaluator.get____())