Aan de slagBegin gratis

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

Bekijk cursus

Oefeninstructies

  • Importeer het vereiste RegressionEvaluator-pakket uit de klasse pyspark.ml.evaluation.
  • Maak de evaluator-code af en geef de metric name op als "rmse". Stel de labelCol in op de naam van de kolom in onze ratings-data die onze ratings bevat (gebruik de methode ratings.columns om kolomnamen te bekijken) en stel de naam van de prediction-kolom in op "prediction".
  • Controleer of de evaluator correct 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____())
Code bewerken en uitvoeren