Freigeben über


SentenceSimilarityTrainer Klasse

Definition

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

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>)

Gilt für: