RLTrainer.h

// Copyright (C) Entropy Software LLC - All Rights Reserved
 
#pragma once
 
#include "Agent.h"
#include "GradientDescentTrainer.h"
#include "Resource.h"
 
namespace SmartEngine
{
 
class IContext;
class IGraphNode;
 
#pragma pack(push, 4)
struct RLTrainerCInfo : public ResourceCInfo
{
    IContext* context = nullptr;
 
    IAgentDataStore* dataStore = nullptr;
 
    const char* agentName = "";
 
    const char** policyNodeNames = nullptr;
 
    int policyNodeNameCount = 0;
 
    float gamma = 0.99f; // Reward discount
 
    GradientDescentTrainingInfo trainingInfo;
 
    int sequenceLength = 1;
};
#pragma pack(pop)
 
class SMARTENGINE_EXPORT IRLTrainer : public virtual IObject, public IAgentFactory, public IResource
{
public:
    SMARTENGINE_DECLARE_CLASS(IRLTrainer)
 
    
    virtual int GetGenerationCount() const = 0;
 
    virtual float GetLoss() = 0;
 
    virtual void Reset() = 0;
 
    virtual void Step() = 0;
};
 
extern "C"
{
    SMARTENGINE_EXPORT int RLTrainer_GetGenerationCount(ObjPtr object);
    SMARTENGINE_EXPORT float RLTrainer_GetLoss(ObjPtr object);
    SMARTENGINE_EXPORT void RLTrainer_Reset(ObjPtr object);
    SMARTENGINE_EXPORT void RLTrainer_Step(ObjPtr object);
    SMARTENGINE_EXPORT ObjPtr RLTrainer_CreateAgent(ObjPtr object);
}
 
} // namespace SmartEngine