Aan de slagBegin gratis

Genereer voorspellingen en bereken de RMSE

Nu we een model hebben dat is getraind op onze gegevens en getuned via cross-validatie, kunnen we kijken hoe het presteert op de test-dataframe. Om dit te doen, berekenen we de RMSE.

Ter info: het genereren van testvoorspellingen duurt met deze gegevensset meer dan een paar minuten. Daarom zijn de testvoorspellingen al gegenereerd en hier beschikbaar als een dataframe genaamd test_predictions. Ter referentie: ze zijn gemaakt met deze code: test_predictions = best_model.transform(test).

Deze oefening maakt deel uit van de cursus

Aanbevelingssystemen bouwen met PySpark

Bekijk cursus

Oefeninstructies

  • De dataframe test_predictions bevat voorspellingen die ons cross-validated ALS-model heeft gegenereerd met de eerder gemaakte test-set. Gebruik de methode .show() om ernaar te kijken en te zien of de voorspellingen in de buurt komen.
  • Gebruik de evaluator die je eerder hebt gebouwd om de RMSE te berekenen door de methode .evaluate() aan te roepen op de gegenereerde test_predictions. Noem dit RMSE.
  • Print de RMSE.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

# View the predictions 
test_predictions.____()

# Calculate and print the RMSE of test_predictions
RMSE = evaluator.____(____)
print(____)
Code bewerken en uitvoeren