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 | 2 つの文の類似性。 |
トレーナーの特性
| 機械学習タスク | 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>) |