leastSquareTestPolicy.h

Go to the documentation of this file.

#ifndef GUM_MULTI_DIM_FUNCTION_GRAPH_LEAST_SQUARE_TEST_POLICY_H
#define GUM_MULTI_DIM_FUNCTION_GRAPH_LEAST_SQUARE_TEST_POLICY_H

// ============================================================================
#include <agrum/FMDP/learning/core/contingencyTable.h>
#include <agrum/FMDP/learning/core/testPolicy/ITestPolicy.h>
// ============================================================================
#include <agrum/multidim/utils/FunctionGraphUtilities/link.h>
// ============================================================================

namespace gum {

  template < typename GUM_SCALAR >
  class LeastSquareTestPolicy : public ITestPolicy< GUM_SCALAR > {
    public:
    // ############################################################################
    // ############################################################################

    // ============================================================================
    // ============================================================================
    LeastSquareTestPolicy() :
        ITestPolicy< GUM_SCALAR >(), __sumO(0.0), __score(0) {
      GUM_CONSTRUCTOR(LeastSquareTestPolicy);
    }

    // ============================================================================
    // ============================================================================
    void* operator new(size_t s) {
      return SmallObjectAllocator::instance().allocate(s);
    }
    void operator delete(void* p) {
      SmallObjectAllocator::instance().deallocate(p,
                                                  sizeof(LeastSquareTestPolicy));
    }

    // ============================================================================
    // ============================================================================
    virtual ~LeastSquareTestPolicy();

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

    // ============================================================================
    // ============================================================================
    void addObservation(Idx attr, GUM_SCALAR value);



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

    // ============================================================================
    // ============================================================================
    bool isTestRelevant() const { return (this->nbObservation() > 20); }



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

    // ============================================================================
    // ============================================================================
    void computeScore();

    // ============================================================================
    // ============================================================================
    double score();

    // ============================================================================
    // ============================================================================
    double secondaryscore() const;



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

    // ============================================================================
    // ============================================================================
    void add(const LeastSquareTestPolicy< GUM_SCALAR >& src);


    // ============================================================================
    // ============================================================================
    double sumValue() const { return __sumO; }

    // ============================================================================
    // ============================================================================
    const HashTable< Idx, Idx >& nbObsTable() const { return __nbObsTable; }

    // ============================================================================
    // ============================================================================
    const HashTable< Idx, double >& sumAttrTable() const { return __sumAttrTable; }

    // ============================================================================
    // ============================================================================
    const HashTable< Idx, LinkedList< double >* >& obsTable() const {
      return __obsTable;
    }

    private:
    double __sumO;

    HashTable< Idx, Idx > __nbObsTable;

    HashTable< Idx, double > __sumAttrTable;

    HashTable< Idx, LinkedList< double >* > __obsTable;

    double __score;
  };

}   // End of namespace gum

#include <agrum/FMDP/learning/core/testPolicy/leastSquareTestPolicy_tpl.h>

#endif /* GUM_MULTI_DIM_FUNCTION_GRAPH_LEAST_SQUARE_TEST_POLICY_H */