[mlpack-git] master: Implemented Save, Load (090f9ba)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Thu Mar 5 21:56:19 EST 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/904762495c039e345beba14c1142fd719b3bd50e...f94823c800ad6f7266995c700b1b630d5ffdcf40
>---------------------------------------------------------------
commit 090f9ba68f981e7216eadaa6442f28ed85c1a0d8
Author: michaelfox99 <michaelfox99 at gmail.com>
Date: Tue Aug 5 13:28:44 2014 +0000
Implemented Save, Load
>---------------------------------------------------------------
090f9ba68f981e7216eadaa6442f28ed85c1a0d8
src/mlpack/core/dists/discrete_distribution.hpp | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/src/mlpack/core/dists/discrete_distribution.hpp b/src/mlpack/core/dists/discrete_distribution.hpp
index 7d7ba9a..7ec605b 100644
--- a/src/mlpack/core/dists/discrete_distribution.hpp
+++ b/src/mlpack/core/dists/discrete_distribution.hpp
@@ -77,7 +77,7 @@ class DiscreteDistribution
/**
* 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 @@ class DiscreteDistribution
*/
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-git
mailing list