시작하기무료로 시작하기

최적 모델과 최적 모델 하이퍼파라미터

이제 교차 검증기 cv를 구성했으니, Spark에 데이터에 ALS 알고리즘을 적합하고 param_grid에 있는 하이퍼파라미터 조합을 시도해 보라고 지시할 수 있어요. 그러면 가장 작은 RMSE를 주는 값을 찾을 수 있습니다. 아쉽게도 여기서는 실행에 시간이 너무 오래 걸리기 때문에 직접 수행하지는 않지만, 방법은 다음과 같습니다:

# 'train' 데이터셋에 교차 검증기 적합
model = cv.fit(train)

# 위의 cv 모델에서 최적 모델 추출
best_model = model.bestModel

이 코드는 별도로 실행되어 best_model이 식별되고 저장되어 있습니다. 제공된 명령을 사용해 모델의 파라미터를 추출하세요.

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

PySpark로 추천 엔진 만들기

강의 보기

연습 안내

  • type(best_model)을 출력해, 하이퍼파라미터 옵션에서 구축된 ALS 모델이 실제로 완성되었는지 확인하세요. 이후의 출력문과 함께 사용하려면 여기서는 print 문이 필요합니다.
  • best_model에서 .getRank() 메서드를 호출해 rank를 추출하세요.
  • best_model에서 .getMaxIter() 메서드를 호출해 maxIter를 추출하세요.
  • best_model에서 .getRegParam() 메서드를 호출해 regParam을 추출하세요.

실습형 인터랙티브 연습

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

# Print best_model
print(____)

# Complete the code below to extract the ALS model parameters
print("**Best Model**")

# Print "Rank"
print("  Rank:", best_model.get____())

# Print "MaxIter"
print("  MaxIter:", best_model.get____())

# Print "RegParam"
print("  RegParam:", best_model.get____())
코드 편집 및 실행