Construiește evaluatorul RMSE
Acum că știi cum să antrenezi un model pe datele de antrenament și să generezi predicții pe datele de test, ai nevoie de o modalitate de a evalua cât de bine performează modelul. În acest scop, vom construi un evaluator. Evaluatorii în Spark pot fi configurați în diverse moduri. Pentru nevoile noastre, dorim un regressionEvaluator care calculează RMSE. După ce construim regressionEvaluator, putem antrena modelul pe datele noastre și genera predicții.
Acest exercițiu face parte din cursul
Construiește motoare de recomandare cu PySpark
Instrucțiuni pentru exercițiu
- Importă pachetul
RegressionEvaluatordin clasapyspark.ml.evaluation. - Completează codul pentru
evaluator, specificândmetric nameca"rmse". SeteazălabelColla numele coloanei din setul de dateratingscare conține ratingurile (folosește metodaratings.columnspentru a vedea numele coloanelor) și setează numele coloanei de predicții la"prediction". - Confirmă că
evaluatora fost creat corect extragând fiecare dintre cei trei parametri din acesta. Fă acest lucru rulând următoarele 3 linii de cod, fiecare în cadrul unuiprint:evaluator.getMetricName()evaluator.getLabelCol()evaluator.getPredictionCol()
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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____())