SentenceSimilarityTrainer Třída

Definice

IEstimator<TTransformer> Představuje pro trénování hluboké neurální sítě (DNN) k určení podobnosti vět.

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

Poznámky

K vytvoření tohoto trenéra použijte TextClassification.

Vstupní a výstupní sloupce

Vstupní data sloupce popisku musí být typu Single a sloupce vět musí být typu TextDataViewType.

Tento trenér vypíše následující sloupce:

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

Charakteristiky trenéra

Charakteristické Value
Úloha strojového učení Regression
Vyžaduje se normalizace? No
Vyžaduje se ukládání do mezipaměti? No
Požadovaný NuGet kromě Microsoft.ML Microsoft.ML.TorchSharp a libtorch-cpu nebo libtorch-cuda-11.3 nebo některé z variant specifických pro operační systém.
Exportovatelné do ONNX No

Podrobnosti trénovacího algoritmu

Trénuje hloubkovou neurální síť (DNN) pomocí existujícího předem natrénovaného modelu NAS-BERT roBERTa pro účely určení podobnosti vět.

Metody

Name Description
Fit(IDataView)

IEstimator<TTransformer> Představuje pro trénování hluboké neurální sítě (DNN) k určení podobnosti vět.

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

IEstimator<TTransformer> Představuje pro trénování hluboké neurální sítě (DNN) k určení podobnosti vět.

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

Metody rozšíření

Name Description
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

Připojte kontrolní bod ukládání do mezipaměti do řetězu odhadce. Tím zajistíte, aby podřízené estimátory byly natrénovány proti datům uloženým v mezipaměti. Před průchodem více dat je užitečné mít kontrolní bod ukládání do mezipaměti.

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

Vzhledem k estimátoru vraťte objekt obtékání, který jednou zavolá delegáta Fit(IDataView) . Často je důležité, aby estimátor vrátil informace o tom, co bylo vhodné, což je důvod, proč Fit(IDataView) metoda vrací konkrétní typ objekt, spíše než jen obecný ITransformer. Ve stejnou dobu se však často vytvářejí do kanálů s mnoha objekty, takže možná budeme muset vytvořit řetěz odhadců prostřednictvím IEstimator<TTransformer> místa, kde odhadovač, EstimatorChain<TLastTransformer> pro který chceme získat transformátor, je někde v tomto řetězu. Pro tento scénář můžeme prostřednictvím této metody připojit delegáta, který bude volána po zavolání fit.

Platí pro