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)

继承: Object

TorchSharpBaseTrainer

TorchSharpBaseTrainer<TLabelCol,TTargetsCol>

NasBertTrainer<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>)