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);
}
}
}