diff --git a/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp b/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp index effe912f22..6ab7817f34 100644 --- a/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp +++ b/src/mlpack/methods/ann/layer/sparse_bias_layer.hpp @@ -136,8 +136,7 @@ class SparseBiasLayer */ template 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 fbb0d67e52..33d34cd2cf 100644 --- a/src/mlpack/methods/ann/layer/sparse_input_layer.hpp +++ b/src/mlpack/methods/ann/layer/sparse_input_layer.hpp @@ -138,8 +138,7 @@ class SparseInputLayer */ template 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 1d5cf70be5..a571210e10 100644 --- a/src/mlpack/methods/ann/layer/sparse_output_layer.hpp +++ b/src/mlpack/methods/ann/layer/sparse_output_layer.hpp @@ -141,19 +141,7 @@ class SparseOutputLayer double Rho() const { return rho; - } - - /** - * Serialize the layer - */ - template - 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; } @@ -184,6 +172,15 @@ class SparseOutputLayer OutputDataType const& Gradient() const { return gradient; } //! Modify the gradient. OutputDataType& Gradient() { return gradient; } + + /** + * Serialize the layer + */ + template + void Serialize(Archive& ar, const unsigned int /* version */) + { + ar & data::CreateNVP(weights, "weights"); + } private: //! Locally-stored number of input units.