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
Övningsinstruktioner
- Importera det
RegressionEvaluator-paket som krävs från klassenpyspark.ml.evaluation. - Fyll i koden för
evaluatoroch angemetric namesom"rmse". SättlabelColtill namnet på den kolumn i våraratings-data som innehåller betygen (använd metodenratings.columnsför att se kolumnnamnen) och sätt kolumnnamnet förpredictiontill"prediction". - Bekräfta att
evaluatorskapades 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____())