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

//
// Author: Roman Kalivoda
//

using System;
using System.Collections.Generic;
using System.IO;
using Microsoft.ML;
using ServerApp.Parser.OutputInfo;
using ServerApp.Parser.Parsers;

namespace ServerApp.Predictor
{
    /// <summary>
    /// Implentation of the <c>IPredicitionController</c> interface.
    /// </summary>
    class PredictionController : IPredictionController
    {
        /// <summary>
        /// A dictionary for storing trained predictors.
        /// </summary>
        private Dictionary<string, int> buildingsToAreas;

        private List<IPredictor> predictors;

        /// <summary>
        /// A reference to a data parser.
        /// </summary>
        private IDataParser dataParser;

        /// <summary>
        /// A feature extractor instance.
        /// </summary>
        private FeatureExtractor featureExtractor;

        /// <summary>
        /// Instantiates new prediction controller.
        /// </summary>
        /// <param name="dataParser">A data parser used to get training data.</param>
        public PredictionController(IDataParser dataParser)
        {
            this.dataParser = dataParser;
            this.predictors = new List<IPredictor>();
            this.buildingsToAreas = new Dictionary<string, int>();
            this.featureExtractor = new FeatureExtractor(dataParser, buildingsToAreas);

            // fill predictors with all available locationKeys
            // TODO Currently all locations use the same predictor. Try dividing locations into subareas with separate predictors.
            var locationKeys = TagInfo.buildings;
            foreach (string key in locationKeys)
            {
                buildingsToAreas.Add(key, 0);
            }
            IPredictor predictor = new NaiveBayesClassifier();
            predictors.Add(predictor);
        }
        public List<string> GetPredictors()
        {
            return new List<string>(buildingsToAreas.Keys);
        }

        public void Load(string locationKey = null, string path = null)
        {
            if (locationKey is null)
            {
                throw new NotImplementedException();
            }
            else
            {
                throw new NotImplementedException();
            }
        }

        public IDataView Predict(string locationKey, WeatherInfo weather, DateTime dateTime)
        {
            IEnumerable<ModelInput> data = new List<ModelInput>
            {
                new ModelInput
                {
                    Temp = (float)weather.temp,
                }
            };
            return this.predictors[buildingsToAreas[locationKey]].Predict(data);
        }


        public void Train(string locationKey = null)
        {
            if (locationKey is null)
            // train all predictors
            {
                // TODO A single predictor is used for all areas, so training is done only once now.
                for (int i = 0; i < this.predictors.Count; i++)
                {
                    // TODO change datetimes when parser interface is ready to parse only downloaded data.
                    //IEnumerable<ModelInput> data = featureExtractor.PrepareModelInput(i, DateTime.MinValue, DateTime.MaxValue);
                    IEnumerable<ModelInput> data = featureExtractor.PrepareModelInput(i, new DateTime(2019, 10, 5), new DateTime(2020, 6, 30));
                    this.predictors[i].Fit(data);
                }
            } else
            // train specified predictor only
            {
                throw new NotImplementedException();
            }
        }


    }
}