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