本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。
文字分類 - TensorFlow 超參數
超參數是在機器學習模型開始學習之前設定的參數。Amazon SageMaker 內建物件偵測 TensorFlow 演算法支援下列超參數。如需有關超參數調校的資訊,請參閱調整文字分類 - TensorFlow 模型。
參數名稱 | 描述 |
---|---|
batch_size |
訓練的批次大小。對於具有多個 的執行個體的訓練GPUs,整個 都會使用此批次大小GPUs。 有效值:正整數。 預設值: |
beta_1 |
有效值:浮點數,範圍:[ 預設值: |
beta_2 |
有效值:浮點數,範圍:[ 預設值: |
dropout_rate |
頂部分類層中退出層的退出率。僅在將 有效值:浮動、範圍:[ 預設值: |
early_stopping |
設定為 有效值:字串,可以是任一:( 預設值: |
early_stopping_min_delta |
符合改善資格所需的變更下限。小於 early_stopping_min_delta 值的絕對變更不符合改善資格。僅在 early_stopping 設為 "True" 時才使用。有效值:浮動、範圍:[ 預設值: |
early_stopping_patience |
在沒有任何改善的情況下,繼續訓練的週期數量。僅在 有效值:正整數。 預設值: |
epochs |
訓練 epoch 的數量。 有效值:正整數。 預設值: |
epsilon |
用於 有效值:浮點數,範圍:[ 預設值: |
initial_accumulator_value |
累加器的起始值或 有效值:浮點數,範圍:[ 預設值: |
learning_rate |
最佳化工具的學習速率。 有效值:浮點數,範圍:[ 預設值: |
momentum |
有效值:浮點數,範圍:[ 預設值: |
optimizer |
最佳化工具類型。如需詳細資訊,請參閱 TensorFlow 文件中的 Optimizer 有效值:字串,下列任何一項:( 預設值: |
regularizers_l2 |
分類層中密集層的 L2 正規化因素。僅在 有效值:浮動、範圍:[ 預設值: |
reinitialize_top_layer |
如果設定為 有效值:字串,下列任一項:( 預設值: |
rho |
有效值:浮點數,範圍:[ 預設值: |
train_only_on_top_layer |
如果 有效值:字串,可以是:( 預設值: |
validation_split_ratio |
要隨機分割以建立驗證資料的訓練資料分數。只有在未透過 有效值:浮點數,範圍:[ 預設值: |
warmup_steps_fraction |
漸層更新步驟總數的分數,其中學習速率從 0 增加到初始學習速率作為暖機。僅與 有效值:浮點數,範圍:[ 預設值: |