bandit_operator.h

Go to the documentation of this file.

#ifndef BANDIT_OPERATOR_H
#define BANDIT_OPERATOR_H
 
#include "../init.h"
// #include "../data/data.h"
#include "../types.h"
#include "../params.h"
#include "../program/tree_node.h"
#include "../ind/fitness.h"
 
namespace Brush {
namespace MAB {
    
class BanditOperator
{
    using Iter = tree<Node>::pre_order_iterator;
public:
    BanditOperator(vector<string> arms);
    
    BanditOperator(map<string, float> arms_probs);
    
    ~BanditOperator() {};

    virtual std::map<string, float> sample_probs(bool update);

    virtual string choose();

    virtual void update(string arm, float reward);
protected:    
    std::map<string, float> probabilities;
};
 
} // MAB
} // Brush
 
#endif // BANDIT_OPERATOR_H