TextClassificationTrainer Classe
Définition
Important
Certaines informations portent sur la préversion du produit qui est susceptible d’être en grande partie modifiée avant sa publication. Microsoft exclut toute garantie, expresse ou implicite, concernant les informations fournies ici.
IEstimator<TTransformer> pour entraîner un réseau neuronal profond (DNN) à classifier du texte.
public class TextClassificationTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<uint,long>type TextClassificationTrainer = class
inherit NasBertTrainer<uint32, int64>Public Class TextClassificationTrainer
Inherits NasBertTrainer(Of UInteger, Long)- Héritage
-
TextClassificationTrainer
Remarques
Pour créer ce formateur, utilisez TextClassification.
Colonnes d’entrée et de sortie
Les données de colonne d’étiquette d’entrée doivent être de type clé et les colonnes de phrase doivent être de typeTextDataViewType.
Ce formateur génère les colonnes suivantes :
| Nom de colonne de sortie | Type de colonne | Description |
|---|---|---|
PredictedLabel |
type de 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é. |
Score |
Vecteur deSingle | Scores de toutes les classes. Une valeur plus élevée signifie une probabilité plus élevée de tomber dans la classe associée. Si l’élément i-th a la plus grande valeur, l’index d’étiquette prédit est i.Notez que i est un index de base zéro. |
Caractéristiques du formateur
| 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é dans le but de classifier du texte.
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>) |