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이어야 합니다.
이 트레이너는 다음 열을 출력합니다.
| 출력 열 이름 | 열 유형 | Description |
|---|---|---|
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>) |