ÎncepețiÎncepe gratuit

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

Vezi cursul

Instrucțiuni pentru exercițiu

  • Importă pachetul RegressionEvaluator din clasa pyspark.ml.evaluation.
  • Completează codul pentru evaluator, specificând metric name ca "rmse". Setează labelCol la numele coloanei din setul de date ratings care conține ratingurile (folosește metoda ratings.columns pentru a vedea numele coloanelor) și setează numele coloanei de predicții la "prediction".
  • Confirmă că evaluator a 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 unui print:
    • 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____())
Editează și rulează codul