avg_backpropagation_strategy.hpp

#ifndef AVG_BACKPROPAGATION_H
#define AVG_BACKPROPAGATION_H

#include "running_stats.hpp"
#include "ibackpropagation_strategy.hpp"

namespace mcts {

// ----------------------------------------------------------------------
// ----------------------------------------------------------------------
class AvgBackpropagationStrategy : public IBackpropagationStrategy {
public:
  AvgBackpropagationStrategy() : stats() {}

  virtual void on_backpropagate(const double &value) { stats.push(value); }

  virtual double ev() const { return stats.mean(); }

  virtual double std_deviation() const { return stats.standard_deviation(); }

  virtual double variance() const { return stats.variance(); }

  virtual int nb_samples() const { return stats.num_data_values(); }

  virtual IBackpropagationStrategy *create() {
    return new AvgBackpropagationStrategy();
  }

private:
  RunningStats stats;
};
}

#endif