SentenceSimilarityTrainer Klasse

Definition

Namespace:

Microsoft.ML.TorchSharp.NasBert

Assembly:

Microsoft.ML.TorchSharp.dll

Paket:

Microsoft.ML.TorchSharp v0.21.1

Die IEstimator<TTransformer> zum Trainieren eines Deep Neural Network (DNN) zum Klassifizieren von Text.

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)

Vererbung

Object

TorchSharpBaseTrainer

TorchSharpBaseTrainer<TLabelCol,TTargetsCol>

NasBertTrainer<Single,Single>

SentenceSimilarityTrainer

Hinweise

Verwenden Sie TextClassification, um diesen Trainer zu erstellen.

Eingabe- und Ausgabespalten

Die Spaltendaten für die Eingabebezeichnung müssen typisiertSingle sein, und die Satzspalten müssen vom TypTextDataViewType sein.

Der Trainer gibt folgende Spalten aus:

Name der Ausgabespalte	Spaltentyp	Beschreibung
Score	Single	Der Grad der Ähnlichkeit zwischen den 2 Sätzen.

Trainermerkmale

ML-Aufgabe	Rregression
Ist eine Normalisierung erforderlich?	Nein
Ist Zwischenspeicherung erforderlich?	Nein
Erforderliche NuGet zusätzlich zu Microsoft.ML	Microsoft.ML.TorchSharp und libtorch-cpu oder libtorch-cuda-11.3 oder eine der betriebssystemspezifischen Varianten.
Exportierbar nach ONNX	Nein

Details des Trainingsalgorithmus

Trainiert ein Deep Neural Network (DNN), indem ein vorhandenes vortrainiertes NAS-BERT RoBERTa-Modell verwendet wird, um die Ähnlichkeit von Sätzen zu bestimmen.

Methoden

Fit(IDataView)	Die IEstimator<TTransformer> zum Trainieren eines Deep Neural Network (DNN) zum Klassifizieren von Text. (Geerbt von TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)	Die IEstimator<TTransformer> zum Trainieren eines Deep Neural Network (DNN) zum Klassifizieren von Text. (Geerbt von NasBertTrainer<TLabelCol,TTargetsCol>)