[mlpack-svn] r16963 - mlpack/trunk/src/mlpack/methods/hmm

[fastlab-svn at coffeetalk-1.cc.gatech.edu]

Author: michaelfox99
Date: Tue Aug  5 09:32:56 2014
New Revision: 16963

Log:
Implemented Save, Load for hierarchical model support


Modified:
   mlpack/trunk/src/mlpack/methods/hmm/hmm_impl.hpp

Modified: mlpack/trunk/src/mlpack/methods/hmm/hmm_impl.hpp
==============================================================================
--- mlpack/trunk/src/mlpack/methods/hmm/hmm_impl.hpp	(original)
+++ mlpack/trunk/src/mlpack/methods/hmm/hmm_impl.hpp	Tue Aug  5 09:32:56 2014
@@ -2,6 +2,7 @@
  * @file hmm_impl.hpp
  * @author Ryan Curtin
  * @author Tran Quoc Long
+ * @author Michael Fox
  *
  * Implementation of HMM class.
  */
@@ -522,6 +523,47 @@
   return convert.str();
 }
 
+//! Save to SaveRestoreUtility
+template<typename Distribution>
+void HMM<Distribution>::Save(util::SaveRestoreUtility& sr) const
+{
+  //  Save parameters.
+  sr.SaveParameter(Type(), "type");
+  sr.SaveParameter(Emission()[0].Type(), "emission_type");
+  sr.SaveParameter(dimensionality, "dimensionality");
+  sr.SaveParameter(transition.n_rows, "states");
+  sr.SaveParameter(transition, "transition");
+  
+  // Now the emissions.
+  util::SaveRestoreUtility mn;
+  for (size_t i = 0; i < transition.n_rows; ++i)
+  {
+    // Generate name.
+    std::stringstream s;
+    s << "emission_distribution_" << i;
+    Emission()[i].Save(mn);
+    sr.AddChild(mn, s.str());
+  }
+}
+
+//! Load from SaveRestoreUtility
+template<typename Distribution>
+void HMM<Distribution>::Load(const util::SaveRestoreUtility& sr)
+{
+  // Load parameters.
+  sr.LoadParameter(dimensionality, "dimensionality");
+  sr.LoadParameter(transition, "transition");
+
+  // Now each emission distribution.
+  Emission().resize(transition.n_rows);
+  for (size_t i = 0; i < transition.n_rows; ++i)
+  {
+    std::stringstream s;
+    s << "emission_distribution_" << i;
+    Emission()[i].Load(sr.Children().at(s.str()));
+  }
+}
+
 }; // namespace hmm
 }; // namespace mlpack