SentenceSimilarityTrainer 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.
Mewakili untuk melatih Deep Neural Network (DNN) untuk menentukan kesamaan IEstimator<TTransformer> kalimat.
public class SentenceSimilarityTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<float,float>
type SentenceSimilarityTrainer = class
inherit NasBertTrainer<single, single>
Public Class SentenceSimilarityTrainer
Inherits NasBertTrainer(Of Single, Single)
- Warisan
-
SentenceSimilarityTrainer
- Warisan
Keterangan
Untuk membuat pelatih ini, gunakan TextClassification.
Kolom input dan output
Data kolom label input harus berjenis Single dan kolom kalimat harus berjenis TextDataViewType.
Pelatih ini menghasilkan kolom berikut:
| Nama kolom output | Jenis kolom | Description |
|---|---|---|
Score |
Single | Tingkat kesamaan antara dua kalimat. |
Karakteristik pelatih
| Karakteristik | Nilai |
|---|---|
| Tugas pembelajaran mesin | Regression |
| 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 ada dan telah dilatih sebelumnya untuk tujuan menentukan kesamaan kalimat.
Metode
| Nama | Deskripsi |
|---|---|
| Fit(IDataView) |
Mewakili untuk melatih Deep Neural Network (DNN) untuk menentukan kesamaan IEstimator<TTransformer> kalimat. (Diperoleh dari NasBertTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
Mewakili untuk melatih Deep Neural Network (DNN) untuk menentukan kesamaan IEstimator<TTransformer> kalimat. (Diperoleh dari NasBertTrainer<TLabelCol,TTargetsCol>) |
Metode Ekstensi
| Nama | Deskripsi |
|---|---|
| AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment) |
Tambahkan 'titik pemeriksaan penembolokan' ke rantai estimator. Ini akan memastikan bahwa estimator hilir akan dilatih terhadap data cache. Sangat membantu untuk memiliki titik pemeriksaan penembolokan sebelum pelatih yang mengambil beberapa data lolos. |
| WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>) |
Mengingat estimator, kembalikan objek pembungkus yang akan memanggil delegasi setelah Fit(IDataView) dipanggil. Seringkali penting bagi estimator untuk mengembalikan informasi tentang apa yang cocok, itulah sebabnya Fit(IDataView) metode mengembalikan objek yang di ketik secara khusus, bukan hanya umum ITransformer. Namun, pada saat yang sama, IEstimator<TTransformer> sering dibentuk menjadi alur dengan banyak objek, jadi kita mungkin perlu membangun rantai estimator melalui EstimatorChain<TLastTransformer> tempat estimator yang ingin kita dapatkan transformator dikubur di suatu tempat dalam rantai ini. Untuk skenario itu, kita dapat melalui metode ini melampirkan delegasi yang akan dipanggil setelah cocok dipanggil. |