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

fastlab-svn at coffeetalk-1.cc.gatech.edu fastlab-svn at coffeetalk-1.cc.gatech.edu
Tue Aug 5 09:28:44 EDT 2014


Author: michaelfox99
Date: Tue Aug  5 09:28:44 2014
New Revision: 16957

Log:
Implemented Save, Load


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

Modified: mlpack/trunk/src/mlpack/core/dists/discrete_distribution.hpp
==============================================================================
--- mlpack/trunk/src/mlpack/core/dists/discrete_distribution.hpp	(original)
+++ mlpack/trunk/src/mlpack/core/dists/discrete_distribution.hpp	Tue Aug  5 09:28:44 2014
@@ -77,7 +77,7 @@
   /**
    * Get the dimensionality of the distribution.
    */
-  size_t Dimensionality() const { return 1; }
+  static size_t const Dimensionality() { return 1; }
 
   /**
    * Return the probability of the given observation.  If the observation is
@@ -144,6 +144,13 @@
    */
   std::string ToString() const;
 
+
+  /** Save to or Load from SaveRestoreUtility
+   */
+  void Save(util::SaveRestoreUtility& n) const;
+  void Load(const util::SaveRestoreUtility& n) { n.LoadParameter(probabilities, "probabilities"); }
+  static std::string const Type() { return "DiscreteDistribution"; }
+    
  private:
   arma::vec probabilities;
 };



More information about the mlpack-svn mailing list