Sdílet prostřednictvím


NerTrainer Třída

Definice

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) pro klasifikaci textu.

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))
Dědičnost

Poznámky

K vytvoření tohoto školitele použijte NER.

Vstupní a výstupní sloupce

Data sloupce vstupního popisku musí být vektor řetězcového typu a sloupce vět musí být typuTextDataViewType .

Tento školitel vypíše následující sloupce:

Název výstupního sloupce Typ sloupce Description
PredictedLabel Vektor typu klíče Index predikovaného popisku Pokud je jeho hodnota i, skutečný popisek by byl i-tý kategorie vstupního popisku s hodnotou klíče.
-- --
Úloha strojového učení Klasifikace s více třídami
Vyžaduje se normalizace? No
Vyžaduje se ukládání do mezipaměti? No
Kromě Microsoft.ML se vyžaduje NuGet. Microsoft.ML.TorchSharp a libtorch-cpu nebo libtorch-cuda-11.3 nebo některou z variant specifických pro operační systém.
Exportovatelné do ONNX No

Podrobnosti trénovacího algoritmu

Trénuje hloubkovou neurální síť (DNN) využitím existujícího předem vytrénovaného modelu nas-BERT roBERTa pro účely rozpoznávání pojmenovaných entit.

Metody

Fit(IDataView)

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) pro klasifikaci textu.

(Zděděno od TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) pro klasifikaci textu.

(Zděděno od NasBertTrainer<TLabelCol,TTargetsCol>)

Platí pro