SentenceSimilarityTrainer Classe

Definição

Representa o IEstimator<TTransformer> treinamento de uma DNN (Rede Neural Profunda) para determinar a similaridade da sentença.

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
Herança
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 tipo Single e as colunas de frase devem ser do tipo TextDataViewType.

Este treinador gera as seguintes colunas:

Nome da coluna de saída Tipo de coluna Descrição
Score Single O grau de semelhança entre as duas frases.

Características do treinador

Característica Value
Tarefa de machine learning Regression
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 existente com a finalidade de determinar a similaridade da sentença.

Métodos

Nome Description
Fit(IDataView)

Representa o IEstimator<TTransformer> treinamento de uma DNN (Rede Neural Profunda) para determinar a similaridade da sentença.

(Herdado de NasBertTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Representa o IEstimator<TTransformer> treinamento de uma DNN (Rede Neural Profunda) para determinar a similaridade da sentença.

(Herdado de NasBertTrainer<TLabelCol,TTargetsCol>)

Métodos de Extensão

Nome Description
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

Acrescente um 'ponto de verificação de cache' à cadeia do avaliador. Isso garantirá que os avaliadores downstream serão treinados em relação aos dados armazenados em cache. É útil ter um ponto de verificação de cache antes dos treinadores que têm várias passagens de dados.

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

Dado um avaliador, retorne um objeto de encapsulamento que chamará um delegado uma vez Fit(IDataView) chamado. Geralmente, é importante que um avaliador retorne informações sobre o que estava em forma, e é por isso que o Fit(IDataView) método retorna um objeto especificamente tipado, em vez de apenas um geral ITransformer. No entanto, ao mesmo tempo, IEstimator<TTransformer> muitas vezes são formados em pipelines com muitos objetos, portanto, talvez seja necessário criar uma cadeia de avaliadores por meio EstimatorChain<TLastTransformer> de onde o avaliador para o qual queremos obter o transformador está enterrado em algum lugar nessa cadeia. Para esse cenário, podemos por meio desse método anexar um delegado que será chamado assim que o ajuste for chamado.

Aplica-se a