SentenceSimilarityTrainer Klasa
Definicja
Ważne
Niektóre informacje odnoszą się do produktu w wersji wstępnej, który może zostać znacząco zmodyfikowany przed wydaniem. Firma Microsoft nie udziela żadnych gwarancji, jawnych lub domniemanych, w odniesieniu do informacji podanych w tym miejscu.
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu.
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)- Dziedziczenie
-
SentenceSimilarityTrainer
Uwagi
Aby utworzyć ten trener, użyj funkcji TextClassification.
Kolumny wejściowe i wyjściowe
Dane kolumny etykiety wejściowej muszą być typem,Single a kolumny zdań muszą być typuTextDataViewType.
Ten trener wyprowadza następujące kolumny:
| Nazwa kolumny wyjściowej | Typ kolumny | Opis |
|---|---|---|
Score |
Single | Stopień podobieństwa między 2 zdaniami. |
Cechy trenera
| Zadanie uczenia maszynowego | Rregression |
| Czy normalizacja jest wymagana? | Nie |
| Czy buforowanie jest wymagane? | Nie |
| Wymagane narzędzie NuGet oprócz Microsoft.ML | Microsoft.ML.TorchSharp i libtorch-cpu lub libtorch-cuda-11.3 lub dowolny z wariantów specyficznych dla systemu operacyjnego. |
| Możliwość eksportowania do ONNX | Nie |
Szczegóły algorytmu trenowania
Trenuje głęboką sieć neuronową (DNN) przy użyciu istniejącego wstępnie wytrenowanego modelu NAS-BERT roBERTa na potrzeby określania podobieństwa zdań.
Metody
| Fit(IDataView) |
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu. (Odziedziczone po TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
Element IEstimator<TTransformer> do trenowania głębokiej sieci neuronowej (DNN) do klasyfikowania tekstu. (Odziedziczone po NasBertTrainer<TLabelCol,TTargetsCol>) |