다음을 통해 공유


ImageClassificationTrainer.Options 클래스

정의

의 옵션 클래스입니다 ImageClassificationTrainer.

public sealed class ImageClassificationTrainer.Options : Microsoft.ML.Trainers.TrainerInputBaseWithLabel
type ImageClassificationTrainer.Options = class
    inherit TrainerInputBaseWithLabel
Public NotInheritable Class ImageClassificationTrainer.Options
Inherits TrainerInputBaseWithLabel
상속
ImageClassificationTrainer.Options

생성자

ImageClassificationTrainer.Options()

의 옵션 클래스입니다 ImageClassificationTrainer.

필드

Arch

전송 학습을 사용하여 이미지 분류 학습의 경우 사용할 모델 아키텍처를 지정합니다. 기본 아키텍처는 Resnet_v2_50.

BatchSize

미니 일괄 처리 학습에 사용할 샘플 수입니다. BatchSize의 기본값은 10입니다.

EarlyStoppingCriteria

학습 메트릭이 개선되지 않을 때 학습을 종료하는 데 사용할 조기 중지 기술 매개 변수입니다. 기본적으로 EarlyStopping이 켜져 있고 모니터링 메트릭은 정확도입니다.

Epoch

학습 반복 횟수입니다. Epoch의 기본값은 200입니다.

FeatureColumnName

기능에 사용할 열입니다.

(다음에서 상속됨 TrainerInputBase)
FinalModelPrefix

그래프 파일을 저장하기 위한 최종 모델 및 검사점 파일/폴더 접두사입니다. 기본 접두사는 "custom_retrained_model_based_on_"입니다.

LabelColumnName

레이블에 사용할 열입니다.

(다음에서 상속됨 TrainerInputBaseWithLabel)
LearningRate

최적화 중에 사용할 학습 속도입니다. 학습 속도의 기본값은 0.01입니다.

LearningRateScheduler

학습 속도 예약을 수행하는 클래스입니다. 기본 학습 속도 스케줄러는 지수 학습 속도 감소입니다.

MetricsCallback

학습 단계 중에 정확도/교차 엔트로피에 대한 통계를 보고하기 위한 콜백입니다. 메트릭 콜백은 기본적으로 null로 설정됩니다.

PredictedLabelColumnName

전송 학습이 완료될 때 마지막 계층의 출력 점수에서 예측된 레이블을 포함할 텐서의 이름입니다. 기본 텐서 이름은 "PredictedLabel"입니다.

ReuseTrainSetBottleneckCachedValues

bin 폴더에서 이미 사용 가능한 경우 캐시된 병목 상태 학습 집합 값을 다시 계산하지 않음을 나타냅니다. 이 매개 변수는 기본적으로 false로 설정됩니다.

ReuseValidationSetBottleneckCachedValues

bin 폴더에서 이미 사용할 수 있는 경우 캐시된 병목 상태 유효성 검사 세트 값을 다시 계산하지 않음을 나타냅니다. 이 매개 변수는 기본적으로 false로 설정됩니다.

ScoreColumnName

전송 학습이 완료될 때 마지막 계층의 출력 점수를 포함할 텐서의 이름입니다. 기본 텐서 이름은 "Score"입니다.

TestOnTrainSet

모든 epoch 이후의 학습 집합에서 모델을 평가하도록 나타냅니다. 학습 세트의 테스트는 기본적으로 true로 설정됩니다.

TrainSetBottleneckCachedValuesFileName

캐싱에 대한 trainset 병목 상태 값을 저장할 작업 영역 내의 파일 이름을 나타내며 기본 파일 이름은 "trainSetBottleneckFile.csv"입니다.

ValidationSet

유효성 검사 집합입니다.

ValidationSetBottleneckCachedValuesFileName

캐싱에 대한 유효성 검사 세트 병목 상태 값을 저장할 작업 영역 내의 파일 이름을 나타내며 기본 파일 이름은 "validationSetBottleneckFile.csv"입니다.

ValidationSetFraction

유효성 검사 집합이 전달되지 않으면 학습 집합의 일부가 유효성 검사로 사용됩니다. 이 동작을 사용하지 않도록 설정하려면 null로 설정합니다 ValidationSetFraction . 0에서 1.0 사이의 값을 허용하고 기본값은 학습 집합의 0.1 또는 10%입니다.

WorkspacePath

이미지 병목 상태 캐시 파일 및 학습된 모델이 저장되는 경로를 나타냅니다. 기본값은 새 임시 디렉터리입니다.

적용 대상