Freigeben über


TextClassificationTrainer Klasse

Definition

Zum IEstimator<TTransformer> Trainieren eines tiefen neuronalen Netzwerks (DNN) zum Klassifizieren von Text.

public class TextClassificationTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<uint,long>
type TextClassificationTrainer = class
    inherit NasBertTrainer<uint32, int64>
Public Class TextClassificationTrainer
Inherits NasBertTrainer(Of UInteger, Long)
Vererbung

Hinweise

Um diesen Trainer zu erstellen, verwenden Sie TextClassification.

Eingabe- und Ausgabespalten

Die Spaltendaten der Eingabebezeichnung müssen schlüsseltyp und die Satzspalten vom TypTextDataViewType sein.

Der Trainer gibt folgende Spalten aus:

Name der Ausgabespalte Spaltentyp Beschreibung
PredictedLabel Typ key Der Index der vorhergesagten Bezeichnung. Wenn sein Wert i ist, wäre die eigentliche Bezeichnung die i-te Kategorie des Typs der Schlüssel-Wert-Eingabebezeichnung.
Score Vektor vonSingle Die Bewertungen aller Klassen. Ein höherer Wert bedeutet eine höhere Wahrscheinlichkeit, in die zugehörige Klasse zu fallen. Wenn das i-te Element den größten Wert hat, wäre der vorhergesagte Bezeichnungsindex i. Beachten Sie, dass „i“ ein nullbasierter Index ist.

Trainereigenschaften

ML-Aufgabe Multiklassenklassifizierung
Ist eine Normalisierung erforderlich? Nein
Ist die Zwischenspeicherung erforderlich? Nein
Erforderlicher 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 zum Trainingsalgorithmus

Trainiert ein Deep Neural Network (DNN), indem ein vorhandenes vortrainiertes NAS-BERT RoBERTa-Modell zum Klassifizieren von Text genutzt wird.

Methoden

Fit(IDataView)

Zum IEstimator<TTransformer> Trainieren eines tiefen neuronalen Netzwerks (DNN) zum Klassifizieren von Text.

(Geerbt von TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Zum IEstimator<TTransformer> Trainieren eines tiefen neuronalen Netzwerks (DNN) zum Klassifizieren von Text.

(Geerbt von NasBertTrainer<TLabelCol,TTargetsCol>)

Gilt für: