SentenceSimilarityTrainer Clase
Definición
Importante
Parte de la información hace referencia a la versión preliminar del producto, que puede haberse modificado sustancialmente antes de lanzar la versión definitiva. Microsoft no otorga ninguna garantía, explícita o implícita, con respecto a la información proporcionada aquí.
IEstimator<TTransformer> para entrenar una red neuronal profunda (DNN) para clasificar 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)- Herencia
-
SentenceSimilarityTrainer
Comentarios
Para crear este instructor, use TextClassification.
Columnas de entrada y salida
Los datos de columna de etiqueta de entrada deben ser de tipoSingle y las columnas de oración deben ser de tipoTextDataViewType .
Este instructor genera las siguientes columnas:
| Nombre de columna de salida | Tipo de columna | Descripción |
|---|---|---|
Score |
Single | Grado de similitud entre las 2 oraciones. |
Características del entrenador
| Tarea de Machine Learning | Rregression |
| ¿Se requiere normalización? | No |
| ¿Se requiere el almacenamiento en caché? | No |
| NuGet necesario además de Microsoft.ML | Microsoft.ML.TorchSharp y libtorch-cpu o libtorch-cuda-11.3 o cualquiera de las variantes específicas del sistema operativo. |
| Exportable a ONNX | No |
Detalles del algoritmo de entrenamiento
Entrena una red neuronal profunda (DNN) aprovechando un modelo NAS-BERT roBERTa previamente entrenado para determinar la similitud de las oraciones.
Métodos
| Fit(IDataView) |
IEstimator<TTransformer> para entrenar una red neuronal profunda (DNN) para clasificar texto. (Heredado de TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
IEstimator<TTransformer> para entrenar una red neuronal profunda (DNN) para clasificar texto. (Heredado de NasBertTrainer<TLabelCol,TTargetsCol>) |