SentenceSimilarityTrainer Classe
Definizione
Importante
Alcune informazioni sono relative alla release non definitiva del prodotto, che potrebbe subire modifiche significative prima della release definitiva. Microsoft non riconosce alcuna garanzia, espressa o implicita, in merito alle informazioni qui fornite.
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à
-
SentenceSimilarityTrainer
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>) |