[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