Sdílet prostřednictvím


SentenceSimilarityTrainer Třída

Definice

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu.

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)
Dědičnost

Poznámky

K vytvoření tohoto školitele použijte TextClassification.

Vstupní a výstupní sloupce

Vstupní data sloupce popisku musí být typeSingle a sloupce vět musí být typuTextDataViewType.

Tento školitel vypíše následující sloupce:

Název výstupního sloupce Typ sloupce Description
Score Single Stupeň podobnosti mezi 2 větami.

Charakteristiky trenéra

Úloha strojového učení Regrese
Vyžaduje se normalizace? No
Vyžaduje se ukládání do mezipaměti? No
Kromě Microsoft.ML vyžaduje NuGet Microsoft.ML.TorchSharp a libtorch-cpu nebo libtorch-cuda-11.3 nebo některou z variant specifických pro operační systém.
Exportovat do ONNX No

Podrobnosti algoritmu trénování

Trénuje hlubokou neurální síť (DNN) využitím existujícího předem vytrénovaného modelu roBERTa NAS-BERT za účelem určení podobnosti vět.

Metody

Fit(IDataView)

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu.

(Zděděno od TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu.

(Zděděno od NasBertTrainer<TLabelCol,TTargetsCol>)

Platí pro