aswi2021tri-musketyri-gitlab/Server/ServerApp/Predictor/NaiveBayesClassifier.cs @ ebe96ca4

//
// Author: Roman Kalivoda
//

using System.Collections.Generic;
using System.Linq;
using Microsoft.ML;
using ServerApp.Parser.OutputInfo;

namespace ServerApp.Predictor
{
    /// <summary>
    /// Implementation of the naive Bayes classifier in ML.NET.
    /// </summary>
    class NaiveBayesClassifier : IPredictor
    {
        /// <summary>
        /// Context of the ML.NET framework.
        /// </summary>
        private MLContext mlContext;

        /// <summary>
        /// Model instance
        /// </summary>
        private ITransformer model;

        /// <summary>
        /// Instantiates new <c>MLContext</c>.
        /// </summary>
        public NaiveBayesClassifier()
        {
            mlContext = new MLContext();
        }

        public void Fit(IEnumerable<ModelInput> trainInput)
        {
            IDataView trainingDataView = mlContext.Data.LoadFromEnumerable(trainInput);
            var dataProcessPipeline = mlContext.Transforms.Conversion.MapValueToKey(nameof(ModelInput.Label))
                .Append(mlContext.Transforms.Concatenate("Features", new[] { "temp" })
                .Append(mlContext.Transforms.NormalizeMinMax("Features", "Features")));
            var trainer = mlContext.MulticlassClassification.Trainers.NaiveBayes();
            var traininingPipeline = dataProcessPipeline.Append(trainer)
                .Append(mlContext.Transforms.Conversion.MapKeyToValue("prediction", "PredictedLabel"));

            this.model = traininingPipeline.Fit(trainingDataView);

        }

        public IDataView Predict(IEnumerable<ModelInput> input)
        {
            var data = mlContext.Data.LoadFromEnumerable(input);
            IDataView result = model.Transform(data);
            return result;
        }
    }
}