[mlpack-git] master: only serialize weights (9b672da)

gitdub at mlpack.org gitdub at mlpack.org
Sat Feb 27 02:53:56 EST 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/a2e57d617e952f1ea2fda8a23e1c6bd2f78beb6d...1bedf15126f6bd0bc93a3233914ac95486a3c0da

>---------------------------------------------------------------

commit 9b672da91c369eab6779c8a7b77ae3f18cff1e0c
Author: stereomatchingkiss <stereomatchingkiss at gmail.com>
Date:   Sat Feb 27 15:53:56 2016 +0800

    only serialize weights


>---------------------------------------------------------------

9b672da91c369eab6779c8a7b77ae3f18cff1e0c
 src/mlpack/methods/ann/layer/sparse_bias_layer.hpp  |  1 -
 src/mlpack/methods/ann/layer/sparse_input_layer.hpp |  1 -
 .../methods/ann/layer/sparse_output_layer.hpp       | 21 +++++++++------------
 3 files changed, 9 insertions(+), 14 deletions(-)

diff --git a/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp b/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp
index effe912..6ab7817 100644
--- a/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp
+++ b/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp
@@ -137,7 +137,6 @@ class SparseBiasLayer
   template<typename Archive>
   void Serialize(Archive& ar, const unsigned int /* version */)
   {    			
-	ar & data::CreateNVP(lambda, "lambda");	
 	ar & data::CreateNVP(weights, "weights");
   }
 
diff --git a/src/mlpack/methods/ann/layer/sparse_input_layer.hpp b/src/mlpack/methods/ann/layer/sparse_input_layer.hpp
index fbb0d67..33d34cd 100644
--- a/src/mlpack/methods/ann/layer/sparse_input_layer.hpp
+++ b/src/mlpack/methods/ann/layer/sparse_input_layer.hpp
@@ -139,7 +139,6 @@ class SparseInputLayer
   template<typename Archive>
   void Serialize(Archive& ar, const unsigned int /* version */)
   {    			
-	ar & data::CreateNVP(lambda, "lambda");	
 	ar & data::CreateNVP(weights, "weights");
   }
 
diff --git a/src/mlpack/methods/ann/layer/sparse_output_layer.hpp b/src/mlpack/methods/ann/layer/sparse_output_layer.hpp
index 1d5cf70..a571210 100644
--- a/src/mlpack/methods/ann/layer/sparse_output_layer.hpp
+++ b/src/mlpack/methods/ann/layer/sparse_output_layer.hpp
@@ -143,18 +143,6 @@ class SparseOutputLayer
     return rho;
   }   
 
-  /**
-   * Serialize the layer
-   */
-  template<typename Archive>
-  void Serialize(Archive& ar, const unsigned int /* version */)
-  {    
-	ar & data::CreateNVP(rhoCap, "rhoCap");		
-	ar & data::CreateNVP(lambda, "lambda");
-	ar & data::CreateNVP(beta, "beta");
-	ar & data::CreateNVP(weights, "weights");
-  }
-
   //! Get the weights.
   OutputDataType const& Weights() const { return weights; }
   //! Modify the weights.
@@ -185,6 +173,15 @@ class SparseOutputLayer
   //! Modify the gradient.
   OutputDataType& Gradient() { return gradient; }
   
+  /**
+   * Serialize the layer
+   */
+  template<typename Archive>
+  void Serialize(Archive& ar, const unsigned int /* version */)
+  {    	
+	ar & data::CreateNVP(weights, "weights");
+  }
+
  private:
   //! Locally-stored number of input units.
   size_t inSize;




More information about the mlpack-git mailing list