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