mlpack  master
Public Member Functions | Private Member Functions | Private Attributes | List of all members
mlpack::ann::ParametersVisitor Class Reference

ParametersVisitor exposes the parameters set of the given module and stores the parameters set into the given matrix. More...

Inheritance diagram for mlpack::ann::ParametersVisitor:
Inheritance graph
[legend]

Public Member Functions

 ParametersVisitor (arma::mat &&parameters)
 Store the parameters set into the given parameters matrix. More...
 
template<typename LayerType >
void operator() (LayerType *layer) const
 Set the parameters set. More...
 

Private Member Functions

template<typename T , typename P >
std::enable_if< !HasParametersCheck< T, P &(T::*)()>::value, void >::type LayerParameters (T *layer, P &output) const
 Do not set the parameters set if the module doesn't implement the Parameters() function. More...
 
template<typename T , typename P >
std::enable_if< HasParametersCheck< T, P &(T::*)()>::value, void >::type LayerParameters (T *layer, P &output) const
 Set the parameters set if the module implements the Parameters() function. More...
 

Private Attributes

arma::mat && parameters
 The parameters set. More...
 

Detailed Description

ParametersVisitor exposes the parameters set of the given module and stores the parameters set into the given matrix.

Definition at line 28 of file parameters_visitor.hpp.

Constructor & Destructor Documentation

mlpack::ann::ParametersVisitor::ParametersVisitor ( arma::mat &&  parameters)

Store the parameters set into the given parameters matrix.

Member Function Documentation

template<typename T , typename P >
std::enable_if< !HasParametersCheck<T, P&(T::*)()>::value, void>::type mlpack::ann::ParametersVisitor::LayerParameters ( T *  layer,
P &  output 
) const
private

Do not set the parameters set if the module doesn't implement the Parameters() function.

template<typename T , typename P >
std::enable_if< HasParametersCheck<T, P&(T::*)()>::value, void>::type mlpack::ann::ParametersVisitor::LayerParameters ( T *  layer,
P &  output 
) const
private

Set the parameters set if the module implements the Parameters() function.

template<typename LayerType >
void mlpack::ann::ParametersVisitor::operator() ( LayerType *  layer) const

Set the parameters set.

Member Data Documentation

arma::mat&& mlpack::ann::ParametersVisitor::parameters
private

The parameters set.

Definition at line 40 of file parameters_visitor.hpp.


The documentation for this class was generated from the following file: