Lingua

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 della colonna di output Tipo di colonna Descrizione
PredictedLabel Vettore di tipo chiave Indice dell'etichetta stimata. Se il valore è i, l'etichetta effettiva sarà la categoria i-th nel tipo di etichetta di input con valori chiave.
-- --
Attività di Machine Learning Classificazione multiclasse
La normalizzazione è necessaria? No
La memorizzazione nella cache è necessaria? No
NuGet necessario 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 sfruttando un modello di NAS-BERT roBERTa con training preliminare esistente allo scopo del riconoscimento di entità denominate.

Metodi

Nome Descrizione
GetOutputSchema(SchemaShape)

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

(Ereditato da NasBertTrainer<TLabelCol,TTargetsCol>)

Metodi di estensione

Nome Descrizione
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

Aggiungere un "checkpoint di memorizzazione nella cache" alla catena di stima. In questo modo si garantisce che gli estimatori downstream vengano sottoposti a training su dati memorizzati nella cache. È utile disporre di un checkpoint di memorizzazione nella cache prima che i formatori eseseguono più passaggi di dati.

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

Dato uno strumento di stima, restituire un oggetto wrapping che chiamerà un delegato una volta Fit(IDataView) chiamato. È spesso importante che uno strumento di stima restituisca informazioni sulle dimensioni, motivo per cui il Fit(IDataView) metodo restituisce un oggetto tipizzato in modo specifico, anziché solo un oggetto generale ITransformer. Tuttavia, allo stesso tempo, IEstimator<TTransformer> vengono spesso formati in pipeline con molti oggetti, quindi potrebbe essere necessario creare una catena di estimatori tramite EstimatorChain<TLastTransformer> dove lo strumento di stima per il quale si vuole ottenere il trasformatore è sepolto in una posizione in questa catena. Per questo scenario, è possibile collegare un delegato che verrà chiamato una volta chiamato fit.

Si applica a