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

//
// Author: Roman Kalivoda
//

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

namespace ServerApp.Predictor
{
    /// <summary>
    /// Interface for PredictionController class. Defines basic interface methods to access the "model" package.
    /// </summary>
    public interface IPredictionController
    {
        /// <summary>
        /// </summary>
        /// <returns>A dictionary with all existing predictors.</returns>
        List<string> GetPredictors();

        /// <summary>
        /// Trains all predictors. The training data are taken from given path location.
        /// </summary>
        /// <param name="locationKey">A string identifier of the location for which to train a predictor.</param>
        void Train(string locationKey = null);

        /// <summary>
        /// Loads trained predictors from files in <c>path</c>.
        /// </summary>
        /// <param name="locationKey">A string identifier of the location for which to load a predictor.</param>
        /// <param name="path">A path to folder with trained prediction model.</param>
        void Load(string locationKey = null,string path = null);

        /// <summary>
        /// Predicts turnout level at given location at given time supposing given weather conditions.
        /// </summary>
        /// <param name="locationKey">String identifier of the location.</param>
        /// <param name="weather">Weather conditions.</param>
        /// <param name="dateTime">Date and time.</param>
        /// <returns></returns>
        IDataView Predict(string locationKey, WeatherInfo weather, DateTime dateTime);
    }
}