Lấy RMSE
Giờ bạn đã biết cách xây dựng mô hình và tạo dự đoán, và có một evaluator để cho biết mô hình dự đoán rating tốt đến đâu, chúng ta có thể tính RMSE để xem mô hình ALS hoạt động ra sao. Ta sẽ dùng evaluator đã xây dựng ở bài trước để tính và in ra rmse.
Bài tập này là một phần của khóa học
Xây dựng Recommendation Engine với PySpark
Hướng dẫn bài tập
- Gọi phương thức
.evaluate()trênevaluatorcủa chúng ta để tính RMSE trên dataframetest_predictions. Gán kết quả vàoRMSE. - In
RMSE
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Evaluate the "test_predictions" dataframe
RMSE = evaluator.____(____)
# Print the RMSE
print (____)