umbraco-ml-train-api.cs

[HttpGet]
[Route("/api/ml/train")]
public IActionResult Train()
{
    var nodes = this.GetData();
    var dataView = context.Data.LoadFromEnumerable<FAQModel>(dataNodes);

    var pipeline = context.Transforms.Conversion.MapValueToKey(outputColumnName: "Label", inputColumnName: "Answer")
        .Append(context.Transforms.Text.FeaturizeText("Features","Question"))
        .Append(context.MulticlassClassification.Trainers.SdcaMaximumEntropy(labelColumnName: "Label", featureColumnName: "Features"))
        .Append(context.Transforms.Conversion.MapKeyToValue(outputColumnName: "PredictedAnswer", inputColumnName: "PredictedLabel")); 

    var model = pipeline.Fit(dataView);
	
    context.Model.Save(model, dataView.Schema, "data/faq-model.zip");

    return Ok($"Trained model based on {dataNodes.Count()} nodes and saved to data/faq-model.zip");
}