시작하기무료로 시작하기

예측 생성 및 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(____)
코드 편집 및 실행