SentenceSimilarityTrainer Classe

Definizione

Rappresenta l'oggetto per il IEstimator<TTransformer> training di una rete neurale profonda (DNN) per determinare la somiglianza delle frasi.

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)
Ereditarietà
Ereditarietà
SentenceSimilarityTrainer

Commenti

Per creare questo formatore, usare TextClassification.

Colonne di input e output

I dati della colonna dell'etichetta di input devono essere di tipo Single e le colonne della frase devono essere di tipo TextDataViewType.

Questo formatore restituisce le colonne seguenti:

Nome colonna di output Tipo di colonna Descrizione
Score Single Grado di somiglianza tra le due frasi.

Caratteristiche del trainer

Caratteristica Valore
Attività di Machine Learning Regression
La normalizzazione è necessaria? No
La memorizzazione nella cache è necessaria? No
NuGet necessario oltre a Microsoft.ML Microsoft.ML.TorchSharp e libtorch-cpu o libtorch-cuda-11.3 o una delle varianti specifiche del sistema operativo.
Esportabile in ONNX No

Dettagli dell'algoritmo di training

Esegue il training di una rete neurale profonda sfruttando un modello di NAS-BERT roBERTa esistente con training preliminare allo scopo di determinare la somiglianza delle frasi.

Metodi

Nome Descrizione
Fit(IDataView)

Rappresenta l'oggetto per il IEstimator<TTransformer> training di una rete neurale profonda (DNN) per determinare la somiglianza delle frasi.

(Ereditato da NasBertTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Rappresenta l'oggetto per il IEstimator<TTransformer> training di una rete neurale profonda (DNN) per determinare la somiglianza delle frasi.

(Ereditato da NasBertTrainer<TLabelCol,TTargetsCol>)

Metodi di estensione

Nome Descrizione
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

Aggiungere un "checkpoint di memorizzazione nella cache" alla catena di stima. In questo modo si garantisce che gli estimatori downstream vengano sottoposti a training su dati memorizzati nella cache. È utile disporre di un checkpoint di memorizzazione nella cache prima che i formatori eseseguono più passaggi di dati.

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

Dato uno strumento di stima, restituire un oggetto wrapping che chiamerà un delegato una volta Fit(IDataView) chiamato. È spesso importante che uno strumento di stima restituisca informazioni sulle dimensioni, motivo per cui il Fit(IDataView) metodo restituisce un oggetto tipizzato in modo specifico, anziché solo un oggetto generale ITransformer. Tuttavia, allo stesso tempo, IEstimator<TTransformer> vengono spesso formati in pipeline con molti oggetti, quindi potrebbe essere necessario creare una catena di estimatori tramite EstimatorChain<TLastTransformer> dove lo strumento di stima per il quale si vuole ottenere il trasformatore è sepolto in una posizione in questa catena. Per questo scenario, è possibile collegare un delegato che verrà chiamato una volta chiamato fit.

Si applica a