random_init.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_PERCEPTRON_INITIALIZATION_METHODS_RANDOM_INIT_HPP
#define MLPACK_METHODS_PERCEPTRON_INITIALIZATION_METHODS_RANDOM_INIT_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace perceptron {
 
class RandomInitialization
{
 public:
  RandomInitialization() { }
 
  inline static void Initialize(arma::mat& weights,
                                arma::vec& biases,
                                const size_t numFeatures,
                                const size_t numClasses)
  {
    weights.randu(numFeatures, numClasses);
    biases.randu(numClasses);
  }
}; // class RandomInitialization
 
} // namespace perceptron
} // namespace mlpack
 
#endif