Kom igångKom igång gratis

Bygg en RMSE-utvärderare

Nu när du vet hur du anpassar en modell till träningsdata och genererar testprediktioner behöver du ett sätt att utvärdera hur väl modellen presterar. För det bygger vi en evaluator. Utvärderare i Spark kan byggas på olika sätt. I vårt fall vill vi ha en regressionEvaluator som beräknar RMSE. När vi har byggt vår regressionEvaluator kan vi anpassa modellen till våra data och generera prediktioner.

Den här övningen är en del av kursen

Bygg rekommendationsmotorer med PySpark

Visa kurs

Övningsinstruktioner

  • Importera det RegressionEvaluator-paket som krävs från klassen pyspark.ml.evaluation.
  • Fyll i koden för evaluator och ange metric name som "rmse". Sätt labelCol till namnet på den kolumn i våra ratings-data som innehåller betygen (använd metoden ratings.columns för att se kolumnnamnen) och sätt kolumnnamnet för prediction till "prediction".
  • Bekräfta att evaluator skapades korrekt genom att hämta var och en av de tre parametrarna från den. Gör detta genom att köra följande 3 kodrader, var och en inuti ett print-uttryck:
    • evaluator.getMetricName()
    • evaluator.getLabelCol()
    • evaluator.getPredictionCol()

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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____())
Redigera och kör kod