Вычисление RMSE
Теперь вы умеете строить модель, генерировать прогнозы и оценивать их качество с помощью evaluator. Пришло время вычислить RMSE, чтобы понять, насколько хорошо справилась модель ALS. Для этого воспользуемся evaluator, созданным в предыдущем упражнении, — вычислим RMSE и выведем результат на экран.
Это упражнение является частью курса
Построение рекомендательных систем с помощью PySpark
Инструкции к упражнению
- Вызовите метод
.evaluate()на объектеevaluator, чтобы вычислить RMSE на датафреймеtest_predictions. Сохраните результат в переменнуюRMSE. - Выведите значение
RMSE.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Evaluate the "test_predictions" dataframe
RMSE = evaluator.____(____)
# Print the RMSE
print (____)