[mlpack-svn] r16956 - mlpack/trunk/src/mlpack/core/dists

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

Author: michaelfox99
Date: Tue Aug  5 09:28:29 2014
New Revision: 16956

Log:
Implemented Save, Load


Modified:
   mlpack/trunk/src/mlpack/core/dists/discrete_distribution.cpp

Modified: mlpack/trunk/src/mlpack/core/dists/discrete_distribution.cpp
==============================================================================
--- mlpack/trunk/src/mlpack/core/dists/discrete_distribution.cpp	(original)
+++ mlpack/trunk/src/mlpack/core/dists/discrete_distribution.cpp	Tue Aug  5 09:28:29 2014
@@ -104,6 +104,14 @@
     probabilities.fill(1 / probabilities.n_elem); // Force normalization.
 }
 
+/** Save to SaveRestoreUtility
+ */
+void DiscreteDistribution::Save(util::SaveRestoreUtility& n) const
+{
+  n.SaveParameter(probabilities, "probabilities");
+  n.SaveParameter(Type(), "type");
+}
+
 /*
  * Returns a string representation of this object.
  */