Partager via


SentenceSimilarityTrainer Classe

Définition

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

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

S’applique à