TextClassificationTrainer Klasa
Definicja
Ważne
Niektóre informacje odnoszą się do produktu w wersji wstępnej, który może zostać znacząco zmodyfikowany przed wydaniem. Firma Microsoft nie udziela żadnych gwarancji, jawnych lub domniemanych, w odniesieniu do informacji podanych w tym miejscu.
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu.
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)- Dziedziczenie
-
TextClassificationTrainer
Uwagi
Aby utworzyć ten trener, użyj funkcji TextClassification.
Kolumny wejściowe i wyjściowe
Dane kolumny etykiety wejściowej muszą być typu klucza , a kolumny zdań muszą być typuTextDataViewType.
Ten trener wyprowadza następujące kolumny:
| Nazwa kolumny wyjściowej | Typ kolumny | Opis |
|---|---|---|
PredictedLabel |
typ klucza | Indeks przewidywanej etykiety. Jeśli jej wartość to i, rzeczywista etykieta będzie kategorią i-th w typie etykiety wejściowej z wartością klucza. |
Score |
WektorSingle | Wyniki wszystkich klas. Wyższa wartość oznacza większe prawdopodobieństwo, że należy do skojarzonej klasy. Jeśli element i-th ma największą wartość, przewidywany indeks etykiety będzie i.Zwróć uwagę, że i jest indeksem zerowym. |
Cechy trenera
| Zadanie uczenia maszynowego | Klasyfikacja wieloklasowa |
| Czy normalizacja jest wymagana? | Nie |
| Czy buforowanie jest wymagane? | Nie |
| Wymagane narzędzie NuGet oprócz Microsoft.ML | Microsoft.ML.TorchSharp i libtorch-cpu lub libtorch-cuda-11.3 lub dowolny z wariantów specyficznych dla systemu operacyjnego. |
| Możliwość eksportowania do ONNX | Nie |
Szczegóły algorytmu trenowania
Trenuje głęboką sieć neuronową (DNN) przy użyciu istniejącego wstępnie wytrenowanego modelu NAS-BERT roBERTa na potrzeby klasyfikowania tekstu.
Metody
| Fit(IDataView) |
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu. (Odziedziczone po TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu. (Odziedziczone po NasBertTrainer<TLabelCol,TTargetsCol>) |