mahalanobis_distance.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_METRICS_MAHALANOBIS_DISTANCE_HPP
#define MLPACK_CORE_METRICS_MAHALANOBIS_DISTANCE_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace metric {
 
template<bool TakeRoot = true>
class MahalanobisDistance
{
 public:
  MahalanobisDistance() { }
 
  MahalanobisDistance(const size_t dimensionality) :
      covariance(arma::eye<arma::mat>(dimensionality, dimensionality)) { }
 
  MahalanobisDistance(arma::mat covariance) :
      covariance(std::move(covariance)) { }
 
  template<typename VecTypeA, typename VecTypeB>
  double Evaluate(const VecTypeA& a, const VecTypeB& b);
 
  const arma::mat& Covariance() const { return covariance; }
 
  arma::mat& Covariance() { return covariance; }
 
  template<typename Archive>
  void serialize(Archive& ar, const unsigned int version);
 
 private:
  arma::mat covariance;
};
 
} // namespace metric
} // namespace mlpack
 
#include "mahalanobis_distance_impl.hpp"
 
#endif