Budowanie ewaluatora RMSE
Wiesz już, jak dopasować model do danych treningowych i generować prognozy na zbiorze testowym. Teraz potrzebujesz sposobu na ocenę jakości modelu. W tym celu zbudujesz evaluator. Ewaluatory w Sparku można tworzyć na różne sposoby. Na nasze potrzeby przyda się regressionEvaluator, który oblicza RMSE. Po jego zbudowaniu możesz dopasować model do danych i wygenerować prognozy.
To ćwiczenie jest częścią kursu
Budowanie silników rekomendacji w PySpark
Instrukcje do ćwiczenia
- Zaimportuj wymagany pakiet
RegressionEvaluatorz klasypyspark.ml.evaluation. - Uzupełnij kod
evaluator, podającmetric namejako"rmse". UstawlabelColna nazwę kolumny w zbiorze danychratings, która zawiera oceny (użyj metodyratings.columns, aby sprawdzić nazwy kolumn), apredictioncolumn name ustaw na"prediction". - Sprawdź, czy
evaluatorzostał poprawnie utworzony, wyodrębniając każdy z trzech parametrów. W tym celu uruchom poniższe 3 linie kodu, każdą w instrukcji print:evaluator.getMetricName()evaluator.getLabelCol()evaluator.getPredictionCol()
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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____())