SentenceSimilarityTrainer Classe
Definição
Importante
Algumas informações se referem a produtos de pré-lançamento que podem ser substancialmente modificados antes do lançamento. A Microsoft não oferece garantias, expressas ou implícitas, das informações aqui fornecidas.
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
-
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>) |