[mlpack-git] master: Add Serialize() to MahalanobisDistance. (bcd1570)

[gitdub at big.cc.gt.atl.ga.us]

Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/4a97187bbba7ce8a6191b714949dd818ef0f37d2...e5905e62c15d1bcff21e6359b11efcd7ab6d7ca0

>---------------------------------------------------------------

commit bcd15705be167f3339e88d86d41ad633a1cd0401
Author: Ryan Curtin <ryan at ratml.org>
Date:   Sat Apr 18 01:37:35 2015 +0000

    Add Serialize() to MahalanobisDistance.


>---------------------------------------------------------------

bcd15705be167f3339e88d86d41ad633a1cd0401
 src/mlpack/core/metrics/mahalanobis_distance.hpp      | 4 ++++
 src/mlpack/core/metrics/mahalanobis_distance_impl.hpp | 9 +++++++++
 2 files changed, 13 insertions(+)

diff --git a/src/mlpack/core/metrics/mahalanobis_distance.hpp b/src/mlpack/core/metrics/mahalanobis_distance.hpp
index 88bcca3..4fcb516 100644
--- a/src/mlpack/core/metrics/mahalanobis_distance.hpp
+++ b/src/mlpack/core/metrics/mahalanobis_distance.hpp
@@ -97,6 +97,10 @@ class MahalanobisDistance
    */
   arma::mat& Covariance() { return covariance; }
 
+  //! Serialize the Mahalanobis distance.
+  template<typename Archive>
+  void Serialize(Archive& ar, const unsigned int version);
+
   //! Return string representation of object.
   std::string ToString() const;
  private:
diff --git a/src/mlpack/core/metrics/mahalanobis_distance_impl.hpp b/src/mlpack/core/metrics/mahalanobis_distance_impl.hpp
index b9f9199..fb42b0e 100644
--- a/src/mlpack/core/metrics/mahalanobis_distance_impl.hpp
+++ b/src/mlpack/core/metrics/mahalanobis_distance_impl.hpp
@@ -42,6 +42,15 @@ double MahalanobisDistance<true>::Evaluate(const VecTypeA& a,
   return sqrt(out[0]);
 }
 
+// Serialize the Mahalanobis distance.
+template<typename TakeRoot>
+template<typename Archive>
+void MahalanobisDistance<TakeRoot>::Serialize(Archive& ar,
+                                              const unsigned int /* version */)
+{
+  ar & data::CreateNVP(covariance, "covariance");
+}
+
 // Convert object into string.
 template<bool TakeRoot>
 std::string MahalanobisDistance<TakeRoot>::ToString() const