Interfaces.cs

using MyCaffe.basecode;
using MyCaffe.basecode.descriptors;
using MyCaffe.common;
using MyCaffe.gym;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.trainers
{
    public enum ITERATOR_TYPE
    {
        ITERATION = 0,
        EPISODE = 1
    }
 
    public interface IXMyCaffeCustomTrainer
    {
        void Initialize(string strProperties, IXMyCaffeCustomTrainerCallback icallback);
        void CleanUp();
        Stage Stage { get; }
        string Name { get; }
        TRAINING_CATEGORY TrainingCategory { get; }
        bool GetUpdateSnapshot(out int nIteration, out double dfAccuracy);
        DatasetDescriptor GetDatasetOverride(int nProjectID, ConnectInfo ci = null);
        bool IsTrainingSupported { get; }
        bool IsTestingSupported { get; }
        bool IsRunningSupported { get; }
        void Train(Component mycaffe, int nIterationOverride, ITERATOR_TYPE type = ITERATOR_TYPE.ITERATION, TRAIN_STEP step = TRAIN_STEP.NONE);
        void Test(Component mycaffe, int nIterationOverride, ITERATOR_TYPE type = ITERATOR_TYPE.ITERATION);
        double GetProperty(string strName);
        string Information { get; }
        void OpenUi();
    }
 
    public interface IXMyCaffeCustomTrainerRL : IXMyCaffeCustomTrainer
    {
        ResultCollection RunOne(Component mycaffe, int nDelay);
        byte[] Run(Component mycaffe, int nN, out string type);
    }
 
    public interface IXMyCaffeCustomTrainerRNN : IXMyCaffeCustomTrainer
    {
        float[] Run(Component mycaffe, int nN);
        byte[] Run(Component mycaffe, int nN, out string type);
        BucketCollection PreloadData(Log log, CancelEvent evtCancel, int nProjectID, PropertySet propertyOverride = null, ConnectInfo ci = null);
        string ResizeModel(Log log, string strModel, BucketCollection rgVocabulary);
    }
 
    public interface IXMyCaffeCustomTrainerCallback
    {
        void Update(TRAINING_CATEGORY cat, Dictionary<string, double> rgValues);
    }
 
    public interface IXMyCaffeCustomTrainerCallbackRNN : IXMyCaffeCustomTrainerCallback
    {
        PropertySet GetRunProperties();
    }
 
    public interface IxTrainer
    {
        bool Initialize();
        bool Shutdown(int nWait);
        bool Train(int nN, ITERATOR_TYPE type, TRAIN_STEP step);
        bool Test(int nN, ITERATOR_TYPE type);
    }
 
    public interface IxTrainerRL : IxTrainer
    {
        ResultCollection RunOne(int nDelay = 1000);
 
        byte[] Run(int nN, PropertySet runProp, out string type);
    }
 
    public interface IxTrainerRNN : IxTrainer
    {
        float[] Run(int nN, PropertySet runProp);
 
        byte[] Run(int nN, PropertySet runProp, out string type);
    }
 
    public interface IxTrainerCallback
    {
        void OnInitialize(InitializeArgs e);
        void OnShutdown();
        void OnGetData(GetDataArgs e);
        void OnUpdateStatus(GetStatusArgs e);
        void OnWait(WaitArgs e);
    }
 
    public interface IxTrainerGetDataCallback
    {
        void OnOverlay(OverlayArgs e);
    }
 
    public interface IxTrainerCallbackRNN : IxTrainerCallback
    {
        void OnConvertOutput(ConvertOutputArgs e);
 
        void OnTestAccuracyUpdate(TestAccuracyUpdateArgs e);
    }
}