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

指示在每个时期之后对训练集评估模型。 默认情况下,对训练集进行测试设置为 true。

TrainSetBottleneckCachedValuesFileName

指示工作区中存储用于缓存的训练集瓶颈值的文件名,默认文件名为“trainSetBottleneckFile.csv”。

ValidationSet

验证集。

ValidationSetBottleneckCachedValuesFileName

指示工作区中存储用于缓存的验证集瓶颈值的文件名,默认文件名为“validationSetBottleneckFile.csv”。

ValidationSetFraction

如果未通过验证集,则将训练集的一部分用作验证。 若要禁用此行为,请将 设置为 ValidationSetFraction null。 接受介于 0 和 1.0 之间的值,默认值为训练集的 0.1 或 10%。

WorkspacePath

指示保存图像瓶颈缓存文件和定型模型的路径,默认为新的临时目录。

适用于