5 references to ImageClassification
Microsoft.ML.AutoML (1)
TrainerExtensions\TrainerExtensionCatalog.cs (1)
44
{ TrainerName.
ImageClassification
, typeof(ImageClassificationExtension) },
Microsoft.ML.AutoML.Tests (2)
TrainerExtensionsTests.cs (2)
34
if (trainerName != TrainerName.
ImageClassification
)
56
var extension = TrainerExtensionCatalog.GetTrainerExtension(TrainerName.
ImageClassification
);
Microsoft.ML.CodeGenerator (2)
CodeGenerator\CSharp\CodeGenerator.cs (1)
34
private static readonly HashSet<string> _imageClassificationTrainers = new HashSet<string>() { TrainerName.
ImageClassification
.ToString() };
CodeGenerator\CSharp\TrainerGeneratorFactory.cs (1)
68
case TrainerName.
ImageClassification
: