kde_stat.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_KDE_STAT_HPP
#define MLPACK_METHODS_KDE_STAT_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace kde {
 
class KDEStat
{
 public:
  KDEStat() :
      mcBeta(0),
      mcAlpha(0),
      accumAlpha(0),
      accumError(0)
  { /* Nothing to do.*/ }
 
  template<typename TreeType>
  KDEStat(TreeType& /* node */) :
      mcBeta(0),
      mcAlpha(0),
      accumAlpha(0),
      accumError(0)
  { /* Nothing to do. */ }
 
  inline double MCBeta() const { return mcBeta; }
 
  inline double& MCBeta() { return mcBeta; }
 
  inline double AccumAlpha() const { return accumAlpha; }
 
  inline double& AccumAlpha() { return accumAlpha; }
 
  inline double AccumError() const { return accumError; }
 
  inline double& AccumError() { return accumError; }
 
  inline double MCAlpha() const { return mcAlpha; }
 
  inline double& MCAlpha() { return mcAlpha; }
 
  template<typename Archive>
  void serialize(Archive& ar, const unsigned int version)
  {
    // Backward compatibility: Old versions of KDEStat needed to handle obsolete
    // values.
    if (version == 0 && Archive::is_loading::value)
    {
      // Placeholders.
      arma::vec centroid;
      bool validCentroid;
 
      ar & BOOST_SERIALIZATION_NVP(centroid);
      ar & BOOST_SERIALIZATION_NVP(validCentroid);
    }
 
    // Backward compatibility: Old versions of KDEStat did not need to handle
    // alpha values.
    if (version > 0)
    {
      ar & BOOST_SERIALIZATION_NVP(mcBeta);
      ar & BOOST_SERIALIZATION_NVP(mcAlpha);
      ar & BOOST_SERIALIZATION_NVP(accumAlpha);
      ar & BOOST_SERIALIZATION_NVP(accumError);
    }
    else if (Archive::is_loading::value)
    {
      mcBeta = -1;
      mcAlpha = -1;
      accumAlpha = -1;
      accumError = -1;
    }
  }
 
 private:
  double mcBeta;
 
  double mcAlpha;
 
  double accumAlpha;
 
  double accumError;
};
 
} // namespace kde
} // namespace mlpack
 
BOOST_TEMPLATE_CLASS_VERSION(template<>, mlpack::kde::KDEStat, 1);
 
#endif