LocalRegression.h

Go to the documentation of this file.

#ifndef LOCALREGRESSION_H_
#define LOCALREGRESSION_H_
 
#include "MUQ/config.h"
 
#if MUQ_HAS_PARCER
#include <parcer/Communicator.h>
#endif
 
#include "MUQ/Modeling/ModPiece.h"
#include "MUQ/Modeling/Flann/FlannCache.h"
 
#include "MUQ/Approximation/Regression/Regression.h"
 
namespace muq {
  namespace Approximation {
    class LocalRegression : public muq::Modeling::ModPiece {
    public:
 
      LocalRegression(std::shared_ptr<muq::Modeling::ModPiece> function, boost::property_tree::ptree& pt);
 
#if MUQ_HAS_PARCER
      LocalRegression(std::shared_ptr<muq::Modeling::ModPiece> function, boost::property_tree::ptree& pt, std::shared_ptr<parcer::Communicator> comm);
#endif
 
#if MUQ_HAS_PARCER
      ~LocalRegression();
#else
      ~LocalRegression() = default;
#endif
 
 
      void Add(std::vector<Eigen::VectorXd> const& inputs) const;
 
 
      Eigen::VectorXd Add(Eigen::VectorXd const& input) const;
 
 
      unsigned int CacheSize() const;
 
 
      bool InCache(Eigen::VectorXd const& point) const;
 
 
      Eigen::VectorXd CachePoint(unsigned int const index) const;
 
 
      std::tuple<Eigen::VectorXd, double, unsigned int> PoisednessConstant(Eigen::VectorXd const& input) const;
 
 
      std::tuple<Eigen::VectorXd, double, unsigned int> PoisednessConstant(Eigen::VectorXd const& input, std::vector<Eigen::VectorXd> const& neighbors) const;
 
 
      std::pair<double, double> ErrorIndicator(Eigen::VectorXd const& input) const;
 
 
      std::pair<double, double> ErrorIndicator(Eigen::VectorXd const& input, std::vector<Eigen::VectorXd> const& neighbors) const;
 
 
      void NearestNeighbors(Eigen::VectorXd const& input, std::vector<Eigen::VectorXd>& neighbors) const;
 
 
      void NearestNeighbors(Eigen::VectorXd const& input, std::vector<Eigen::VectorXd>& neighbors, std::vector<Eigen::VectorXd>& result) const;
 
      Eigen::VectorXd EvaluateRegressor(Eigen::VectorXd const& input, std::vector<Eigen::VectorXd> const& neighbors, std::vector<Eigen::VectorXd> const& result) const;
 
#if MUQ_HAS_PARCER
      void Probe() const;
#endif
 
      const unsigned int kn;
 
 
      Eigen::VectorXd CacheCentroid() const;
 
      void ClearCache();
 
      unsigned int Order() const;
 
    private:
 
      virtual void EvaluateImpl(muq::Modeling::ref_vector<Eigen::VectorXd> const& inputs) override;
 
      void FitRegression(Eigen::VectorXd const& input) const;
 
      void SetUp(std::shared_ptr<muq::Modeling::ModPiece> function, boost::property_tree::ptree& pt);
 
      std::shared_ptr<muq::Modeling::FlannCache> cache;
 
      std::shared_ptr<Regression> reg;
 
#if MUQ_HAS_PARCER
      std::shared_ptr<parcer::Communicator> comm = nullptr;
      const int tagSingle = 0;
#endif
    };
  } // namespace Approximation
} // namespace muq
 
#endif