SentenceSimilarityTrainer Classe

Definição

Namespace:

Microsoft.ML.TorchSharp.NasBert

Assembly:

Microsoft.ML.TorchSharp.dll

Pacote:

Microsoft.ML.TorchSharp v0.21.1

O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto.

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)

Herança

Object

TorchSharpBaseTrainer

TorchSharpBaseTrainer<TLabelCol,TTargetsCol>

NasBertTrainer<Single,Single>

SentenceSimilarityTrainer

Comentários

Para criar esse treinador, use TextClassification.

Colunas de entrada e saída

Os dados da coluna do rótulo de entrada devem ser do tipoSingle e as colunas de frase devem ser do tipoTextDataViewType .

Este treinador gera as seguintes colunas:

Nome da Coluna de Saída	Tipo de coluna	Descrição
Score	Single	O grau de similaridade entre as duas frases.

Características do Treinador

Ferramenta de machine learning	Rregression
A normalização é necessária?	No
O cache é necessário?	No
NuGet necessário além de Microsoft.ML	Microsoft.ML.TorchSharp e libtorch-cpu ou libtorch-cuda-11.3 ou qualquer uma das variantes específicas do sistema operacional.
Exportável para ONNX	No

Detalhes do algoritmo de treinamento

Treina uma DNN (Rede Neural Profunda) aproveitando um modelo roBERTa NAS-BERT pré-treinado existente com a finalidade de determinar a similaridade da frase.

Métodos

Fit(IDataView)	O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto. (Herdado de TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)	O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto. (Herdado de NasBertTrainer<TLabelCol,TTargetsCol>)