public async Task <IActionResult> AddSample([FromServices] SpamDetectorDbContext context, string message, bool isSpam)
{
context.SpamRecords.Add(new SpamRecord()
{
Message = message, IsSpam = isSpam
});
await context.SaveChangesAsync();
return(Ok());
}
public async Task <IActionResult> RemoveSample([FromServices] SpamDetectorDbContext context, int id)
{
var spamRecord = await context.SpamRecords.SingleOrDefaultAsync(x => x.Id == id);
if (spamRecord == null)
{
return(NotFound($"Spam record with id {id} not found."));
}
context.SpamRecords.Remove(spamRecord);
await context.SaveChangesAsync();
return(Ok());
}
public async Task Save(ITransformer bestRunModel, MLContext mlContext, DataViewSchema schema, MulticlassClassificationMetrics metrics)
{
await using var ms = new MemoryStream();
mlContext.Model.Save(bestRunModel, schema, ms);
var trainedModel = new TrainedModel()
{
ModelData = ms.ToArray(),
Accuracy = metrics.MicroAccuracy
};
_context.TrainedModels.Add(trainedModel);
await _context.SaveChangesAsync();
}
