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____())