Condividi tramite


SentenceSimilarityTrainer Classe

Definizione

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

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à

Commenti

Per creare questo formatore, usare TextClassification.

Colonne di input e output

I dati della colonna dell'etichetta di input devono essere di tipoSingle e le colonne della frase devono essere di tipoTextDataViewType .

Questo formatore restituisce le colonne seguenti:

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

Caratteristiche del formatore

Attività di Machine Learning Rregression
La normalizzazione è necessaria? No
La memorizzazione nella cache è necessaria? No
NuGet richiesto 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 algoritmo di training

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

Metodi

Fit(IDataView)

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

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

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

(Ereditato da NasBertTrainer<TLabelCol,TTargetsCol>)

Si applica a