#include <shark/Unsupervised/RBM/BinaryRBM.h>
#include <shark/Algorithms/GradientDescent/SteepestDescent.h>
#include <shark/Unsupervised/RBM/Problems/BarsAndStripes.h>
#include <shark/Unsupervised/RBM/analytics.h>
#include <iostream>
Go to the source code of this file.
Functions | |
void | initializeWeights (BinaryRBM &rbm) |
int | main () |
void initializeWeights | ( | BinaryRBM & | rbm | ) |
Definition at line 16 of file BinaryRBM.cpp.
References shark::RBM< VisibleLayerT, HiddenLayerT, RngT >::numberOfParameters(), shark::RBM< VisibleLayerT, HiddenLayerT, RngT >::setParameterVector(), and shark::uni().
Referenced by main().
int main | ( | ) |
Definition at line 24 of file BinaryRBM.cpp.
References shark::BarsAndStripes::data(), shark::SteepestDescent::init(), initializeWeights(), shark::BarsAndStripes::inputDimension(), shark::negativeLogLikelihood(), shark::ContrastiveDivergence< Operator >::setData(), shark::ContrastiveDivergence< Operator >::setK(), shark::SteepestDescent::setLearningRate(), shark::SteepestDescent::setMomentum(), shark::RBM< VisibleLayerT, HiddenLayerT, RngT >::setStructure(), and shark::SteepestDescent::step().