|
using Microsoft.ML.SearchSpace;
using OptionAttribute = Microsoft.ML.SearchSpace.NestOptionAttribute;
using ColorsOrder = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorsOrder;
using ColorBits = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorBits;
using ResizingKind = Microsoft.ML.Transforms.Image.ImageResizingEstimator.ResizingKind;
using Anchor = Microsoft.ML.Transforms.Image.ImageResizingEstimator.Anchor;
using BertArchitecture = Microsoft.ML.TorchSharp.NasBert.BertArchitecture;
using static Microsoft.ML.Vision.ImageClassificationTrainer.Architecture;
using DataKind = Microsoft.ML.Data.DataKind;
using CaseMode = Microsoft.ML.Transforms.Text.TextNormalizingEstimator.CaseMode;
#nullable enable
namespace Microsoft.ML.AutoML.CodeGen
{
public class ImageClassificationOption
{
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public string LabelColumnName {get; set;} = "Label";
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public string ScoreColumnName {get; set;} = "Score";
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public string FeatureColumnName {get; set;} = "Feature";
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public Microsoft.ML.Vision.ImageClassificationTrainer.Architecture Arch {get; set;} = ResnetV250;
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public int BatchSize {get; set;} = 10;
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public int Epoch {get; set;} = 200;
}
}
|