public void AzureCodeGeneratorTest() { (var pipeline, var columnInference, var mapping) = GetMockedAzurePipelineAndInference(); var setting = new CodeGeneratorSettings() { TrainDataset = @"\path\to\file", ModelPath = @"\path\to\model", MlTask = TaskKind.MulticlassClassification, OutputName = @"test", OutputBaseDir = @"\path\to\test", LabelName = "label", Target = GenerateTarget.ModelBuilder, StablePackageVersion = "StablePackageVersion", UnstablePackageVersion = "UnstablePackageVersion", OnnxModelPath = @"\path\to\onnx", IsAzureAttach = true, IsImage = false, OnnxInputMapping = mapping, }; var codeGen = new AzureAttachCodeGenenrator(pipeline, columnInference, setting); foreach (var project in codeGen.ToSolution()) { foreach (var projectFile in project) { NamerFactory.AdditionalInformation = projectFile.Name; Approvals.Verify(((ICSharpFile)projectFile).File); } } }
public void AzureImageCodeGeneratorTest() { (var pipeline, var columnInference) = GetMockedAzureImagePipelineAndInference(); var setting = new CodeGeneratorSettings() { TrainDataset = @"/path/to/dataset", ModelPath = @"/path/to/model", MlTask = TaskKind.MulticlassClassification, OutputName = @"CodeGenTest", OutputBaseDir = @"/path/to/codegen", LabelName = "Label", Target = GenerateTarget.ModelBuilder, StablePackageVersion = "stableversion", UnstablePackageVersion = "unstableversion", OnnxModelPath = @"/path/to/onnxModel", IsAzureAttach = true, IsImage = true, ClassificationLabel = new string[] { "label1", "label2", "label3" }, }; var codeGen = new AzureAttachCodeGenenrator(pipeline, columnInference, setting); foreach (var project in codeGen.ToSolution()) { foreach (var projectFile in project) { NamerFactory.AdditionalInformation = projectFile.Name; Approvals.Verify(((ICSharpFile)projectFile).File); } } }