Partager via


NerTrainer Classe

Définition

IEstimator<TTransformer> pour entraîner un réseau neuronal profond (DNN) à classifier du texte.

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))
Héritage

Remarques

Pour créer ce formateur, utilisez NER.

Colonnes d’entrée et de sortie

Les données de colonne d’étiquette d’entrée doivent être un vecteur de type chaîne et les colonnes de phrase doivent être de typeTextDataViewType.

Cet entraîneur génère les colonnes suivantes :

Nom de colonne de sortie Type de colonne Description
PredictedLabel Vecteur de type clé Index de l’étiquette prédite. Si sa valeur est i, l’étiquette réelle est la i-ème catégorie dans le type d’étiquette d’entrée avec une valeur de clé.
-- --
Tâche d’apprentissage automatique Classification multiclasse
La normalisation est-elle requise ? No
La mise en cache est-elle obligatoire ? No
NuGet requis en plus de Microsoft.ML Microsoft.ML.TorchSharp et libtorch-cpu ou libtorch-cuda-11.3 ou l’une des variantes spécifiques du système d’exploitation.
Exportable vers ONNX No

Détails de l’algorithme d’entraînement

Forme un réseau neuronal profond (DNN) en tirant parti d’un modèle nas-BERT roBERTa préentraîné à des fins de reconnaissance d’entité nommée.

Méthodes

Fit(IDataView)

IEstimator<TTransformer> pour entraîner un réseau neuronal profond (DNN) à classifier du texte.

(Hérité de TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

IEstimator<TTransformer> pour entraîner un réseau neuronal profond (DNN) à classifier du texte.

(Hérité de NasBertTrainer<TLabelCol,TTargetsCol>)

S’applique à