treeRegress.h

Go to the documentation of this file.

#ifndef GUM_TREE_REGRESS_H
#define GUM_TREE_REGRESS_H

#include <functional>

#include <agrum/multidim/implementations/multiDimFunctionGraph.h>
#include <agrum/multidim/utils/FunctionGraphUtilities/terminalNodePolicies/ExactTerminalNodePolicy.h>

namespace gum {

  template < typename GUM_SCALAR,
             template < typename >
             class COMBINEOPERATOR,
             template < typename >
             class PROJECTOPERATOR,
             template < typename > class TerminalNodePolicy =
                ExactTerminalNodePolicy >
  class TreeRegress {
    public:
    // ============================================================================
    // ============================================================================

    TreeRegress(
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* qAction,
       const Bijection<
          const DiscreteVariable*,
          const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* > pxi);

    ~TreeRegress();

    // ============================================================================
    // ============================================================================

    MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* compute();


    private:
    MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >*
       __xPloreVFunc(NodeId currentNodeId);

    const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* __vFunc;
    const Bijection<
       const DiscreteVariable*,
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* >
                                              __pxi;
    HashTable< const DiscreteVariable*, Idx > __context;
  };

#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
  extern template class TreeRegress< double, std::multiplies, std::plus >;
#endif

}   // namespace gum

#include <agrum/multidim/utils/FunctionGraphUtilities/operators/treeRegress_tpl.h>

#endif   // GUM_REGRESS_H