Condividi tramite


NerTrainer Classe

Definizione

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

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

Commenti

Per creare questo formatore, usare NER.

Colonne di input e output

I dati della colonna dell'etichetta di input devono essere di tipo Vector e le colonne della frase devono essere di tipoTextDataViewType .

Questo formatore restituisce le colonne seguenti:

Nome colonna di output Tipo di colonna Descrizione
PredictedLabel Vettore del tipo di chiave L'indice dell'etichetta stimata. Se il valore è i, l'etichetta effettiva potrebbe essere la categoria i-esima nel tipo di etichetta di input con valori key.
-- --
Attività di Machine Learning Classificazione multiclasse
La normalizzazione è necessaria? No
La memorizzazione nella cache è necessaria? No
NuGet richiesto oltre a Microsoft.ML Microsoft.ML.TorchSharp e libtorch-cpu o libtorch-cuda-11.3 o una delle varianti specifiche del sistema operativo.
Esportabile in ONNX No

Dettagli algoritmo di training

Esegue il training di una rete neurale profonda (DNN) sfruttando un modello NAS-BERT roBERTa con training preliminare esistente allo scopo del riconoscimento di entità denominate.

Metodi

Fit(IDataView)

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

(Ereditato da TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Oggetto IEstimator<TTransformer> per il training di una rete neurale profonda (DNN) per classificare il testo.

(Ereditato da NasBertTrainer<TLabelCol,TTargetsCol>)

Si applica a