SentenceSimilarityTrainer 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 SentenceSimilarityTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<float,float>
type SentenceSimilarityTrainer = class
inherit NasBertTrainer<single, single>
Public Class SentenceSimilarityTrainer
Inherits NasBertTrainer(Of Single, Single)
- Héritage
-
SentenceSimilarityTrainer
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 typeSingle 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 |
---|---|---|
Score |
Single | Degré de similarité entre les 2 phrases. |
Caractéristiques du formateur
Tâche d’apprentissage automatique | Irrégression |
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 déterminer la similarité des phrases.
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>) |