시작하기무료로 시작하기

RMSE 평가기 만들기

이제 학습 데이터에 모델을 적합하고 테스트 예측을 생성하는 방법을 알았으니, 모델이 얼마나 잘 동작하는지 평가할 방법이 필요해요. 이를 위해 evaluator를 만들어 보겠습니다. Spark의 evaluator는 여러 방식으로 만들 수 있어요. 여기서는 RMSE를 계산하는 regressionEvaluator가 필요합니다. regressionEvaluator를 만든 뒤, 모델을 데이터에 적합하고 예측을 생성해 보겠습니다.

이 연습은 강의의 일부입니다

PySpark로 추천 엔진 만들기

강의 보기

연습 안내

  • pyspark.ml.evaluation 클래스에서 필요한 RegressionEvaluator 패키지를 임포트하세요.
  • evaluator 코드를 완성해 metric name"rmse"로 지정하세요. ratings 데이터에서 평점이 들어 있는 열 이름을 labelCol에 설정하고(열 이름은 ratings.columns 메서드로 확인), 예측 열 이름은 "prediction"으로 설정하세요.
  • evaluator가 올바르게 생성되었는지 세 가지 파라미터를 각각 추출해 확인하세요. 다음 세 줄을 각각 print 문으로 실행합니다:
    • evaluator.getMetricName()
    • evaluator.getLabelCol()
    • evaluator.getPredictionCol()

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Import RegressionEvaluator
from pyspark.ml.evaluation import ____

# Complete the evaluator code
evaluator = RegressionEvaluator(metricName="____", labelCol="____", predictionCol="____")

# Extract the 3 parameters
print(evaluator.get____())
print(evaluator.get____())
print(evaluator.get____())
코드 편집 및 실행