SentenceSimilarityTrainer Класс
Определение
Важно!
Некоторые сведения относятся к предварительной версии продукта, в которую до выпуска могут быть внесены существенные изменения. Майкрософт не предоставляет никаких гарантий, явных или подразумеваемых, относительно приведенных здесь сведений.
Для IEstimator<TTransformer> обучения глубокой нейронной сети (DNN) для классификации текста.
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)- Наследование
-
SentenceSimilarityTrainer
Комментарии
Чтобы создать этот модуль обучения, используйте TextClassification.
Входные и выходные столбцы
Данные столбца входных меток должны быть типамиSingle , а столбцы предложений должны иметь типTextDataViewType .
Этот алгоритм обучения выводит следующие столбцы:
| Имя выходного столбца | Тип столбца | Описание |
|---|---|---|
Score |
Single | Степень сходства между двумя предложениями. |
Характеристики инструктора
| Задача машинного обучения | Rregression |
| Требуется ли нормализация? | Нет |
| Требуется ли кэширование? | Нет |
| Обязательный NuGet в дополнение к Microsoft.ML | Microsoft.ML.TorchSharp и libtorch-cpu или libtorch-cuda-11.3 или любой из вариантов операционной системы. |
| Возможность экспорта в ONNX | Нет |
Сведения об алгоритме обучения
Обучает глубокую нейронную сеть (DNN), используя существующую предварительно обученную модель ROBERTa NAS-BERT для определения сходства предложений.
Методы
| Fit(IDataView) |
Для IEstimator<TTransformer> обучения глубокой нейронной сети (DNN) для классификации текста. (Унаследовано от TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
Для IEstimator<TTransformer> обучения глубокой нейронной сети (DNN) для классификации текста. (Унаследовано от NasBertTrainer<TLabelCol,TTargetsCol>) |