使用 SageMaker AI 自動調校模型 - Amazon SageMaker AI

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

使用 SageMaker AI 自動調校模型

Amazon SageMaker AI 自動模型調校 (AMT) 透過在資料集上執行許多訓練任務,找到模型的最佳版本。Amazon SageMaker AI 自動模型調校 (AMT) 也稱為超參數調校。為此,AMT 會使用您指定的演算法和超參數範圍。它接著會根據您選擇的指標,選擇可讓模型取得最佳執行結果的超參數值。

例如,在行銷資料集上執行二進位分類問題。您的目標是透過訓練 搭配 Amazon SageMaker AI 的 XGBoost 演算法 模型,將演算法曲線指標下的面積 (AUC) 最大化。建議您尋找可訓練最佳模型的 etaalphamin_child_weightmax_depth 超參數值。指定這些超參數值的範圍。然後,SageMaker AI 超參數調校會在範圍內搜尋,以尋找建立訓練任務的組合,以建立具有最高 AUC 的模型。為了節省資源或滿足特定模型品質期望,請設定完成條件,以在符合條件後停止調校。

您可以將 SageMaker AI AMT 與內建演算法、自訂演算法或 SageMaker AI 預先建置容器搭配使用,以用於機器學習架構。

SageMaker AI AMT 可以使用 Amazon EC2 Spot 執行個體來最佳化執行訓練任務時的成本。如需詳細資訊,請參閱Amazon SageMaker AI 中的受管 Spot 訓練

開始使用超參數調校之前,您應該擁有明確定義的機器學習問題,包括:

  • 資料集

  • 了解您要訓練的演算法類型

  • 清楚了解如何衡量是否成功

準備您的資料集和演算法,讓它們在 SageMaker AI 中運作,並至少成功執行一次訓練任務。如需設定和執行訓練任務的詳細資訊,請參閱Amazon SageMaker AI 設定指南