adaptiveRMaxPlaner.h

Go to the documentation of this file.

// =========================================================================
#ifndef GUM_ADAPTIVE_RMAX_PLANER_H
#define GUM_ADAPTIVE_RMAX_PLANER_H
// =========================================================================
#include <agrum/FMDP/SDyna/Strategies/IDecisionStrategy.h>
#include <agrum/FMDP/fmdp.h>
#include <agrum/FMDP/learning/fmdpLearner.h>
#include <agrum/FMDP/planning/structuredPlaner.h>
#include <agrum/FMDP/simulation/statesCounter.h>
// =========================================================================

namespace gum {

  class AdaptiveRMaxPlaner
      : public StructuredPlaner< double >
      , public IDecisionStrategy {
    // ###################################################################
    // ###################################################################
    public:
    // ==========================================================================
    // ==========================================================================
    static AdaptiveRMaxPlaner*
       ReducedAndOrderedInstance(const ILearningStrategy* learner,
                                 double                   discountFactor = 0.9,
                                 double                   epsilon = 0.00001,
                                 bool                     verbose = true) {
      return new AdaptiveRMaxPlaner(new MDDOperatorStrategy< double >(),
                                    discountFactor,
                                    epsilon,
                                    learner,
                                    verbose);
    }

    // ==========================================================================
    // ==========================================================================
    static AdaptiveRMaxPlaner* TreeInstance(const ILearningStrategy* learner,
                                            double discountFactor = 0.9,
                                            double epsilon = 0.00001,
                                            bool   verbose = true) {
      return new AdaptiveRMaxPlaner(new TreeOperatorStrategy< double >(),
                                    discountFactor,
                                    epsilon,
                                    learner,
                                    verbose);
    }


    // ###################################################################
    // ###################################################################
    private:
    // ==========================================================================
    // ==========================================================================
    AdaptiveRMaxPlaner(IOperatorStrategy< double >* opi,
                       double                       discountFactor,
                       double                       epsilon,
                       const ILearningStrategy*     learner,
                       bool                         verbose);

    // ==========================================================================
    // ==========================================================================
    public:
    ~AdaptiveRMaxPlaner();



    // ###################################################################
    // ###################################################################

    public:
    // ==========================================================================
    // ==========================================================================
    void initialize(const FMDP< double >* fmdp);


    // ==========================================================================
    // ==========================================================================
    void makePlanning(Idx nbStep = 1000000);



    // ###################################################################
    // ###################################################################

    protected:
    // ==========================================================================
    // ==========================================================================
    virtual void _initVFunction();

    // ==========================================================================
    // ==========================================================================
    virtual MultiDimFunctionGraph< double >* _valueIteration();



    // ###################################################################
    // ###################################################################

    protected:
    // ==========================================================================
    // ==========================================================================
    virtual void _evalPolicy();


    private:
    void __makeRMaxFunctionGraphs();

    std::pair< NodeId, NodeId > __visitLearner(const IVisitableGraphLearner*,
                                               NodeId currentNodeId,
                                               MultiDimFunctionGraph< double >*,
                                               MultiDimFunctionGraph< double >*);
    void                        __clearTables();

    private:
    HashTable< Idx, MultiDimFunctionGraph< double >* > __actionsRMaxTable;
    HashTable< Idx, MultiDimFunctionGraph< double >* > __actionsBoolTable;
    const ILearningStrategy*                           __fmdpLearner;

    double __rThreshold;
    double __rmax;


    // ###################################################################
    // ###################################################################
    public:
    void checkState(const Instantiation& newState, Idx actionId) {
      if (!__initializedTable[actionId]) {
        __counterTable[actionId]->reset(newState);
        __initializedTable[actionId] = true;
      } else
        __counterTable[actionId]->incState(newState);
    }

    private:
    HashTable< Idx, StatesCounter* > __counterTable;
    HashTable< Idx, bool >           __initializedTable;

    bool __initialized;
  };

} /* namespace gum */

#endif   // GUM_ADAPTIVE_RMAX_PLANER_H