NerTrainer 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 NerTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<Microsoft.ML.Data.VBuffer<uint>,Microsoft.ML.Data.VBuffer<long>>type NerTrainer = class
inherit NasBertTrainer<VBuffer<uint32>, VBuffer<int64>>Public Class NerTrainer
Inherits NasBertTrainer(Of VBuffer(Of UInteger), VBuffer(Of Long))- Dziedziczenie
Uwagi
Aby utworzyć ten trener, użyj narzędzia NER.
Kolumny wejściowe i wyjściowe
Dane kolumny etykiety wejściowej muszą być wektorem typu ciągu , a kolumny zdania muszą być typuTextDataViewType.
Ten trener wyprowadza następujące kolumny:
| Nazwa kolumny wyjściowej | Typ kolumny | Opis |
|---|---|---|
PredictedLabel |
Wektor typu 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. |
| -- | -- | |
| 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 rozpoznawania nazwanych jednostek.
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>) |