TextClassificationTrainer 클래스
정의
중요
일부 정보는 릴리스되기 전에 상당 부분 수정될 수 있는 시험판 제품과 관련이 있습니다. Microsoft는 여기에 제공된 정보에 대해 어떠한 명시적이거나 묵시적인 보증도 하지 않습니다.
IEstimator<TTransformer> 텍스트를 분류하는 DNN(심층 신경망)을 학습하기 위한 입니다.
public class TextClassificationTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<uint,long>type TextClassificationTrainer = class
inherit NasBertTrainer<uint32, int64>Public Class TextClassificationTrainer
Inherits NasBertTrainer(Of UInteger, Long)- 상속
-
TextClassificationTrainer
설명
이 트레이너를 만들려면 TextClassification을 사용합니다.
입력 및 출력 열
입력 레이블 열 데이터는 키 형식이어야 하며 문장 열은 형식TextDataViewType이어야 합니다.
이 트레이너는 다음 열을 출력합니다.
| 출력 열 이름 | 열 유형 | Description |
|---|---|---|
PredictedLabel |
키 형식 | 예측된 레이블의 인덱스입니다. 값이 i라면 실제 레이블은 키 값 입력 레이블 형식에서 i번째 범주입니다. |
Score |
의 벡터Single | 모든 클래스의 점수입니다. 값이 높을수록 연결된 클래스에 해당할 가능성이 높습니다. i번째 요소의 값이 가장 큰 경우 예측된 레이블 인덱스는 i가 됩니다. i는 0부터 시작하는 인덱스입니다. |
트레이너 특성
| 기계 학습 작업 | 다중 클래스 분류 |
| 정규화가 필요한가요? | 예 |
| 캐싱이 필요한가요? | 예 |
| 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>) |