View a markdown version of this page

Antarmuka input dan output untuk Klasifikasi Teks - TensorFlow algoritma - Amazon SageMaker AI
Pelatihan tambahanInferensi dengan Klasifikasi Teks - algoritma TensorFlow

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

Antarmuka input dan output untuk Klasifikasi Teks - TensorFlow algoritma

Setiap model pra-terlatih yang tercantum dalam Model TensorFlow Hub dapat disesuaikan ke kumpulan data apa pun yang terdiri dari kalimat teks dengan sejumlah kelas. Model pra-latih melampirkan lapisan klasifikasi ke model Penyematan Teks dan menginisialisasi parameter lapisan ke nilai acak. Dimensi keluaran lapisan klasifikasi ditentukan berdasarkan jumlah kelas yang terdeteksi dalam data input.

Perhatikan cara memformat data pelatihan Anda untuk masukan ke Klasifikasi Teks - TensorFlow model.

  • Format input data pelatihan: Direktori yang berisi data.csv file. Setiap baris kolom pertama harus memiliki label kelas integer antara 0 dan jumlah kelas. Setiap baris kolom kedua harus memiliki data teks yang sesuai.

Berikut ini adalah contoh file CSV masukan. Perhatikan bahwa file tidak boleh memiliki header apa pun. File harus di-host di bucket Amazon S3 dengan jalur yang mirip dengan berikut ini:s3://bucket_name/input_directory/. Perhatikan bahwa trailing / diperlukan.

|   |  |
|---|---|
|0 |hide new secretions from the parental units|
|0 |contains no wit , only labored gags|
|1 |that loves its characters and communicates something rather beautiful about human nature|
|...|...|

Pelatihan tambahan

Anda dapat menyemai pelatihan model baru dengan artefak dari model yang Anda latih sebelumnya dengan SageMaker AI. Pelatihan tambahan menghemat waktu pelatihan ketika Anda ingin melatih model baru dengan data yang sama atau serupa.

catatan

Anda hanya dapat menyemai Klasifikasi Teks SageMaker AI - TensorFlow model dengan Klasifikasi Teks lain - TensorFlow model yang dilatih dalam SageMaker AI.

Anda dapat menggunakan dataset apa pun untuk pelatihan tambahan, selama kumpulan kelas tetap sama. Langkah pelatihan inkremental mirip dengan langkah penyempurnaan, tetapi alih-alih memulai dengan model yang telah dilatih sebelumnya, Anda mulai dengan model yang disetel dengan baik yang ada.

Untuk informasi lebih lanjut tentang menggunakan pelatihan tambahan dengan Klasifikasi Teks SageMaker AI - TensorFlow algoritma, lihat buku catatan sampel Peng antar JumpStart - Klasifikasi Teks.

Inferensi dengan Klasifikasi Teks - algoritma TensorFlow

Anda dapat meng-host model yang disetel dengan baik yang dihasilkan dari pelatihan Klasifikasi TensorFlow Teks Anda untuk inferensi. Setiap format teks mentah untuk inferensi harus tipe application/x-text konten.

Menjalankan inferensi menghasilkan nilai probabilitas, label kelas untuk semua kelas, dan label prediksi yang sesuai dengan indeks kelas dengan probabilitas tertinggi yang dikodekan dalam format JSON. Klasifikasi Teks - TensorFlow model memproses satu string per permintaan dan menghasilkan hanya satu baris. Berikut ini adalah contoh respons format JSON:

accept: application/json;verbose

{"probabilities": [prob_0, prob_1, prob_2, ...],
"labels": [label_0, label_1, label_2, ...],
"predicted_label": predicted_label}

Jika accept disetel keapplication/json, maka model hanya mengeluarkan probabilitas.