Bouw een RMSE-evaluator
Nu je weet hoe je een model op trainingsdata fit en testvoorspellingen genereert, heb je een manier nodig om te beoordelen hoe goed je model presteert. Daarvoor bouwen we een evaluator. Evaluators in Spark kun je op verschillende manieren opzetten. Voor ons doel willen we een regressionEvaluator die de RMSE berekent. Nadat we onze regressionEvaluator hebben gebouwd, kunnen we het model fitten op onze data en voorspellingen genereren.
Deze oefening maakt deel uit van de cursus
Aanbevelingssystemen bouwen met PySpark
Oefeninstructies
- Importeer het vereiste
RegressionEvaluator-pakket uit de klassepyspark.ml.evaluation. - Maak de
evaluator-code af en geef demetric nameop als"rmse". Stel delabelColin op de naam van de kolom in onzeratings-data die onze ratings bevat (gebruik de methoderatings.columnsom kolomnamen te bekijken) en stel de naam van deprediction-kolom in op"prediction". - Controleer of de
evaluatorcorrect is aangemaakt door elk van de drie parameters eruit te halen. Doe dit door de volgende 3 regels code uit te voeren, elk binnen een print-statement:evaluator.getMetricName()evaluator.getLabelCol()evaluator.getPredictionCol()
Interactieve oefening met praktijkervaring
Probeer deze oefening door deze voorbeeldcode aan te vullen.
# 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____())