[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 ¶meters, arma::mat &output)
+void MaximalInputs(arma::mat const ¶meters, 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 ¶meters, 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 ¶meters, arma::mat &output);
+void MaximalInputs(arma::mat const ¶meters, arma::mat &output,
+ double minRange = 0, double maxRange = 255);
} // namespace nn
} // namespace mlpack
More information about the mlpack-git
mailing list