예측 생성 및 RMSE 계산
이제 데이터로 학습하고 교차 검증으로 튜닝한 모델이 준비되었으니, test 데이터프레임에서 성능을 확인해 보겠습니다. 이를 위해 RMSE를 계산할 거예요.
참고로, 이 데이터셋에서는 테스트 예측을 생성하는 데 몇 분 이상 걸립니다. 그래서 이미 생성된 테스트 예측을 test_predictions라는 데이터프레임으로 제공해 드립니다. 참고용으로, 다음 코드로 생성되었습니다: test_predictions = best_model.transform(test).
이 연습은 강의의 일부입니다
PySpark로 추천 엔진 만들기
연습 안내
- 데이터프레임
test_predictions에는 이전에 만든test세트를 사용해 교차 검증된 ALS 모델이 생성한 예측이 담겨 있어요..show()메서드를 사용해 확인하고, 예측이 꽤 근접해 보이는지 살펴보세요. - 이전에 만들었던
evaluator를 사용해, 생성된test_predictions에 대해.evaluate()메서드를 호출하여RMSE를 계산하세요. 이 값을RMSE라고 하세요. RMSE를 출력하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# View the predictions
test_predictions.____()
# Calculate and print the RMSE of test_predictions
RMSE = evaluator.____(____)
print(____)