SentenceSimilarityTrainer 类
定义
重要
一些信息与预发行产品相关，相应产品在发行之前可能会进行重大修改。 对于此处提供的信息，Microsoft 不作任何明示或暗示的担保。
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 |
| 是否需要规范化？ | 否 |
| 是否需要缓存？ | 否 |
| 除 Microsoft.ML 外，还需要 NuGet | Microsoft.ML.TorchSharp 和 libtorch-cpu 或 libtorch-cuda-11.3 或任何特定于 OS 的变体。 |
| 可导出到 ONNX | 否 |
训练算法详细信息
利用现有的预先训练的 NAS-BERT roBERTa 模型来确定句子相似性，训练深度神经网络 (DNN) 。
方法
| Fit(IDataView) |
IEstimator<TTransformer>用于训练深度神经网络的 (DNN) 对文本进行分类。 (继承自 TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
IEstimator<TTransformer>用于训练深度神经网络的 (DNN) 对文本进行分类。 (继承自 NasBertTrainer<TLabelCol,TTargetsCol>) |