ImageClassificationTrainer.Options 类
定义
重要
一些信息与预发行产品相关,相应产品在发行之前可能会进行重大修改。 对于此处提供的信息,Microsoft 不作任何明示或暗示的担保。
public sealed class ImageClassificationTrainer.Options : Microsoft.ML.Trainers.TrainerInputBaseWithLabel
type ImageClassificationTrainer.Options = class
inherit TrainerInputBaseWithLabel
Public NotInheritable Class ImageClassificationTrainer.Options
Inherits TrainerInputBaseWithLabel
- 继承
构造函数
ImageClassificationTrainer.Options() |
字段
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 |
指示保存图像瓶颈缓存文件和定型模型的路径,默认为新的临时目录。 |