/// <summary/>
public void ConfigureServices(IServiceCollection services)
{
services.AddControllers();
services.AddSwaggerDocument(config =>
{
config.GenerateExamples = true;
config.PostProcess = document =>
{
document.Info.Version = "v1";
document.Info.Title = "Simple ML API";
document.Info.Description = "Cat or dog on your image???";
};
});
// Подготовка программы к запуску
var solutionDirectory = Path.GetFullPath(Path.Combine(AppContext.BaseDirectory,
"..",
"..",
"..",
"..",
".."));
var resoursesInfo = new ResoursesInfo(solutionDirectory);
var mlContext = new MLContext();
// Обучение модели
var engine = new BinaryClassifierEngine(resoursesInfo, mlContext);
var trainedModel = engine.Learning();
services.AddSingleton(resoursesInfo);
services.AddSingleton(mlContext);
services.AddSingleton(trainedModel);
services.AddScoped <ClassifyService>();
}
/// <summary/>
public ClassifyService(
MLContext context,
ITransformer trainedModel,
ResoursesInfo resourses)
{
_context = context;
_trainedModel = trainedModel;
_resourses = resourses;
}
/// <summary/>
public BinaryClassifierEngine(ResoursesInfo resourses, MLContext context)
{
_resourses = resourses;
_context = context;
}