aswi2021tri-musketyri-gitlab/Server/ServerApp/Predictor/IPredictor.cs @ 662b2404

//
// Author: Roman Kalivoda
//

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Microsoft.ML;
using Microsoft.ML.Data;


namespace ServerApp.Predictor
{
    /// <summary>
    /// A predictor interface.
    /// </summary>
    public interface IPredictor
    {
        /// <summary>
        /// Trains the predictor with the given training data input.
        /// </summary>
        /// <param name="trainInput">A collection of <c>ModelInput</c> instances. The objects contain both feature vector inputs and corresponding labels.</param>
        void Fit(IEnumerable<ModelInput> trainInput);

        /// <summary>
        /// Predicts classes to the given feature vectors.
        /// </summary>
        /// <param name="input">A collection of model feature vectors in <c>ModelInput</c> instances.</param>
        /// <returns>A collection of <c>PredictionResult</c> instances.</returns>
        IDataView Predict(IEnumerable<ModelInput> input);

        // TODO define Save method
    }
}