LossTrainer.h

// Copyright (C) Entropy Software LLC - All Rights Reserved
 
#pragma once
 
#include "Context.h"
#include "Graph.h"
#include "Object.h"
#include "Resource.h"
 
namespace SmartEngine
{
 
#pragma pack(push, 4)
struct LossTrainerCInfo : public ResourceCInfo
{
    IContext* context = nullptr;
 
    IGraph* graph = nullptr;
};
 
enum class LossTrainingMethod : byte
{
    WholeDataset,
 
    Stochastic,
};
 
struct LossTrainingMethodInfo
{
    LossTrainingMethod method = LossTrainingMethod::Stochastic;
 
    const int64* sessionIndices = nullptr;
 
    int sessionCount = 0;
 
    int batchSize = 32;
 
    int sequenceLength = 1;
};
#pragma pack(pop)
 
class SMARTENGINE_EXPORT ILossTrainer : public virtual IObject, public IResource
{
public:
    SMARTENGINE_DECLARE_CLASS(ILossTrainer)
 
    
    virtual void SetTrainingMethod(const LossTrainingMethodInfo& info) = 0;
 
    virtual float GetLoss() = 0;
 
    virtual uint32 GetGenerationCount() const = 0;
};
 
extern "C"
{
    SMARTENGINE_EXPORT void LossTrainer_SetTrainingMethod(ObjPtr object, const LossTrainingMethodInfo& info);
    SMARTENGINE_EXPORT float LossTrainer_GetLoss(ObjPtr object);
    SMARTENGINE_EXPORT uint32 LossTrainer_GetGenerationCount(ObjPtr object);
}
 
} // namespace SmartEngine