LoslegenKostenlos starten

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

Übungsanweisungen

  • Importiere das benötigte Paket RegressionEvaluator aus der Klasse pyspark.ml.evaluation.
  • Vervollständige den evaluator-Code und setze den metric name auf "rmse". Setze labelCol auf den Namen der Spalte in unseren ratings-Daten, die die Bewertungen enthält (verwende ratings.columns, um die Spaltennamen zu sehen), und setze den Namen der prediction-Spalte auf "prediction".
  • Bestätige, dass der evaluator korrekt 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____())
Code bearbeiten und ausführen