[mlpack-git] master: can adjust the output range (e4294dd)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Mon Nov 30 10:40:09 EST 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/5aaf0e441dd64a5de9a0210aa7a837eecf162d12...e4519fc42a2a340cf0387ab082bf49b9715c871b

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

commit e4294dd97cf6a4e3eba1f1d9d12b0849dedbf253
Author: stereomatchingkiss <stereomatchingkiss at gmail.com>
Date:   Wed Oct 28 15:31:08 2015 +0800

    can adjust the output range


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

e4294dd97cf6a4e3eba1f1d9d12b0849dedbf253
 src/mlpack/methods/sparse_autoencoder/maximal_inputs.cpp | 4 ++--
 src/mlpack/methods/sparse_autoencoder/maximal_inputs.hpp | 5 ++++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/mlpack/methods/sparse_autoencoder/maximal_inputs.cpp b/src/mlpack/methods/sparse_autoencoder/maximal_inputs.cpp
index b3217e4..bc2a8a6 100644
--- a/src/mlpack/methods/sparse_autoencoder/maximal_inputs.cpp
+++ b/src/mlpack/methods/sparse_autoencoder/maximal_inputs.cpp
@@ -44,7 +44,7 @@ void VisualizeHiddenUnit(size_t rows, size_t cols,
 
 }
 
-void MaximalInputs(arma::mat const &parameters, arma::mat &output)
+void MaximalInputs(arma::mat const &parameters, arma::mat &output, double minRange, double maxRange)
 {
   arma::mat paramTemp(parameters.t());
   double const mean = arma::mean(arma::mean(paramTemp));
@@ -68,7 +68,7 @@ void MaximalInputs(arma::mat const &parameters, arma::mat &output)
   double const max = output.max();
   double const min = output.min();
   if((max - min) != 0) {
-    output = (output - min) / (max - min) * 255;
+    output = (output - min) / (max - min) * (maxRange - minRange) + minRange;
   }
 }
 
diff --git a/src/mlpack/methods/sparse_autoencoder/maximal_inputs.hpp b/src/mlpack/methods/sparse_autoencoder/maximal_inputs.hpp
index cfa7926..dbefd02 100644
--- a/src/mlpack/methods/sparse_autoencoder/maximal_inputs.hpp
+++ b/src/mlpack/methods/sparse_autoencoder/maximal_inputs.hpp
@@ -14,6 +14,8 @@ namespace nn {
  * http://deeplearning.stanford.edu/wiki/index.php/Exercise:Sparse_Autoencoder
  * @param params The parameters want to maximize
  * @param output Parameters after maximize
+ * @param minRange minimum range of the output, default value is 0
+ * @param maxRange maximum range of the output, default value is 255
  * @pre 1 : The layout of the parameters should be same as following
  * //          vSize   1
  * //       |        |  |
@@ -44,7 +46,8 @@ namespace nn {
  * maximalInput.save("trained.pgm", arma::pgm_binary);
  * @endcode
  */
-void MaximalInputs(arma::mat const &parameters, arma::mat &output);
+void MaximalInputs(arma::mat const &parameters, arma::mat &output,
+                   double minRange = 0, double maxRange = 255);
 
 } // namespace nn
 } // namespace mlpack



More information about the mlpack-git mailing list