TextClassificationTrainer Kelas
Definisi
Penting
Beberapa informasi terkait produk prarilis yang dapat diubah secara signifikan sebelum dirilis. Microsoft tidak memberikan jaminan, tersirat maupun tersurat, sehubungan dengan informasi yang diberikan di sini.
IEstimator<TTransformer> untuk melatih Deep Neural Network (DNN) untuk mengklasifikasikan teks.
public class TextClassificationTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<uint,long>type TextClassificationTrainer = class
inherit NasBertTrainer<uint32, int64>Public Class TextClassificationTrainer
Inherits NasBertTrainer(Of UInteger, Long)- Warisan
-
TextClassificationTrainer
Keterangan
Untuk membuat pelatih ini, gunakan TextClassification.
Kolom Input dan Output
Data kolom label input harus berupa jenis kunci dan kolom kalimat harus berjenisTextDataViewType.
Pelatih ini menghasilkan kolom berikut:
| Nama Kolom Output | Jenis Kolom | Deskripsi |
|---|---|---|
PredictedLabel |
jenis kunci | Indeks label yang diprediksi. Jika nilainya adalah i, label aktual akan menjadi kategori i-th dalam jenis label input bernilai kunci. |
Score |
Vektor dariSingle | Skor semua kelas. Nilai yang lebih tinggi berarti probabilitas yang lebih tinggi untuk masuk ke kelas terkait. Jika elemen i-th memiliki nilai terbesar, indeks label yang diprediksi adalah i.Perhatikan bahwa i adalah indeks berbasis nol. |
Karakteristik Pelatih
| Tugas pembelajaran mesin | Klasifikasi multikelas |
| Apakah normalisasi diperlukan? | Tidak |
| Apakah penembolokan diperlukan? | Tidak |
| NuGet yang diperlukan selain Microsoft.ML | Microsoft.ML.TorchSharp dan libtorch-cpu atau libtorch-cuda-11.3 atau salah satu varian spesifik OS. |
| Dapat diekspor ke ONNX | Tidak |
Detail Algoritma Pelatihan
Melatih Deep Neural Network (DNN) dengan memanfaatkan model ROBERTa NAS-BERT yang sudah dilatih sebelumnya untuk tujuan mengklasifikasikan teks.
Metode
| Fit(IDataView) |
IEstimator<TTransformer> untuk melatih Deep Neural Network (DNN) untuk mengklasifikasikan teks. (Diperoleh dari TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
IEstimator<TTransformer> untuk melatih Deep Neural Network (DNN) untuk mengklasifikasikan teks. (Diperoleh dari NasBertTrainer<TLabelCol,TTargetsCol>) |