SentenceSimilarityTrainer Kelas

Definisi

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
Warisan
SentenceSimilarityTrainer

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.

Berlaku untuk