reinforce_config.h

Go to the documentation of this file.

#ifndef REINFORCE_CONFIG_H
#define REINFORCE_CONFIG_H
 
#include "cuberl/base/cuberl_types.h"
#include "bitrl/bitrl_consts.h"
#include "cuberl/utils/train_enum_type.h"
 
#include <ostream>
#include <string>
 
namespace cuberl {
namespace rl {
namespace algos {
namespace pg {
 
enum class BaselineEnumType { NONE=-1, CONSTANT=0, MEAN=1, STANDARDIZE=2};
    
 
struct ReinforceConfig
{
    bool normalize_rewards{false};
    
    cuberl::utils::TrainEnumType train_type{cuberl::utils::TrainEnumType::BATCH};
    
    BaselineEnumType baseline_type{BaselineEnumType::NONE};
    
    DeviceType device_type;
    
    uint_t n_episodes;
 
    uint_t max_itrs_per_episode;
    
    real_t gamma;
    
    real_t baseline_constant{0.0};
    
    real_t eps{bitrl::consts::TOLERANCE};
    
 
    std::ostream& print(std::ostream& out)const;
    
    void load_from_json(const std::string& filename);
};
 
 
inline
std::ostream& operator<<(std::ostream& out, ReinforceConfig opts){
    return opts.print(out);
}
 
}
}
}
}
 
#endif