View a markdown version of this page

Menyetel Klasifikasi Teks - TensorFlow model - Amazon SageMaker AI

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Menyetel Klasifikasi Teks - TensorFlow model

Penyetelan model otomatis, juga dikenal sebagai penyetelan hyperparameter, menemukan versi model terbaik dengan menjalankan banyak pekerjaan yang menguji berbagai hyperparameter pada dataset Anda. Anda memilih hyperparameter yang dapat disetel, rentang nilai untuk masing-masing, dan metrik objektif. Anda memilih metrik objektif dari metrik yang dihitung algoritma. Penyetelan model otomatis mencari hyperparameter yang dipilih untuk menemukan kombinasi nilai yang menghasilkan model yang mengoptimalkan metrik objektif.

Untuk informasi selengkapnya tentang penyetelan model, lihatPenyetelan model otomatis dengan SageMaker AI.

Metrik dihitung oleh Klasifikasi Teks - TensorFlow algoritma

Lihat bagan berikut untuk menemukan metrik mana yang dihitung oleh Klasifikasi Teks - TensorFlow algoritma.

Nama Metrik Deskripsi Arah Optimasi Pola Regex
validation:accuracy

Rasio jumlah prediksi yang benar dengan jumlah total prediksi yang dibuat.

Maksimalkan

val_accuracy=([0-9\\.]+)

Klasifikasi Teks yang Dapat Disetel - hyperparameter TensorFlow

Setel model klasifikasi teks dengan hyperparameter berikut. Hyperparameter yang memiliki dampak terbesar pada metrik tujuan klasifikasi teks adalah:batch_size,learning_rate, danoptimizer. Setel hyperparameter terkait pengoptimal, seperti,,momentum, regularizers_l2 beta_1beta_2, dan eps berdasarkan yang dipilih. optimizer Misalnya, gunakan beta_1 dan beta_2 hanya ketika adamw atau adam apakahoptimizer.

Untuk informasi lebih lanjut tentang hyperparameter mana yang digunakan untuk masing-masingoptimizer, lihatKlasifikasi Teks - TensorFlow Hyperparameter.

Nama Parameter Jenis Parameter Rentang yang Direkomendasikan
batch_size

IntegerParameterRanges

MinValue: 4, MaxValue: 128

beta_1

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0,999

beta_2

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0,999

eps

ContinuousParameterRanges

MinValue: 1e-8, MaxValue: 1.0

learning_rate

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0,5

momentum

ContinuousParameterRanges

MinValue: 0,0, MaxValue: 0,999

optimizer

CategoricalParameterRanges

['adamw', 'adam', 'sg', 'rmsprop', 'nesterov', 'adagrad', 'adadelta']

regularizers_l2

ContinuousParameterRanges

MinValue: 0,0, MaxValue: 0,999

train_only_on_top_layer

CategoricalParameterRanges

['Benar', 'Salah']