episode_info.h

Go to the documentation of this file.

#ifndef EPISODE_INFO_H
#define EPISODE_INFO_H
 
#include "cuberl/base/cuberl_types.h"
#include "bitrl/bitrl_consts.h"
#include <chrono>
#include <string>
#include <map>
#include <any>
#include <ostream>
 
namespace cuberl {
namespace rl {
 
struct EpisodeInfo
{
 
    uint_t episode_index{bitrl::consts::INVALID_ID};
    uint_t episode_iterations{bitrl::consts::INVALID_ID};
    bool stop_training{false};
 
    real_t episode_reward{0.0};
 
    std::chrono::duration<real_t> total_time;
 
    std::map<std::string, std::any> info;
 
    std::ostream& print(std::ostream& out)const noexcept;
};
 
inline
std::ostream& operator<<(std::ostream& out, const EpisodeInfo& info){
    return info.print(out);
}
 
}
 
}
 
#endif // EPISODE_INFO_H