jens
6ad74f4ee7
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@560 b431acfa-c32f-4a4a-93f1-934dc6c82436
78 lines
1.5 KiB
C++
78 lines
1.5 KiB
C++
/*
|
|
* To change this license header, choose License Headers in Project Properties.
|
|
* To change this template file, choose Tools | Templates
|
|
* and open the template in the editor.
|
|
*/
|
|
|
|
/*
|
|
* File: Rbm.hpp
|
|
* Author: jens
|
|
*
|
|
* Created on 21. Oktober 2019, 21:28
|
|
*/
|
|
|
|
#ifndef RBM_HPP
|
|
#define RBM_HPP
|
|
|
|
#include <armadillo>
|
|
#include "noise.h"
|
|
class IRbmListener
|
|
{
|
|
public:
|
|
|
|
virtual ~IRbmListener()
|
|
{
|
|
|
|
}
|
|
|
|
bool onProgress()
|
|
{
|
|
return true;
|
|
}
|
|
};
|
|
class Rbm
|
|
{
|
|
public:
|
|
struct Params
|
|
{
|
|
Params()
|
|
: m_weightDecay(0.01)
|
|
, m_learningRate(0.1)
|
|
, m_momentum(0.5)
|
|
, m_doRaoBlackwell(true)
|
|
, m_doSampleVisible(false)
|
|
, m_doSampleBatch(false)
|
|
, m_numGibbs(1)
|
|
{
|
|
}
|
|
|
|
double m_weightDecay;
|
|
double m_learningRate;
|
|
double m_momentum;
|
|
bool m_doRaoBlackwell;
|
|
bool m_doSampleVisible;
|
|
bool m_doSampleBatch;
|
|
size_t m_numGibbs;
|
|
};
|
|
|
|
Rbm(size_t numHidden, size_t numVisible);
|
|
Rbm(const Rbm& orig);
|
|
virtual ~Rbm();
|
|
|
|
void train(arma::mat const &batch, size_t numEpochs, size_t sizeMiniBatch, Params const ¶ms, IRbmListener *pListener);
|
|
arma::mat sample(arma::mat const &src);
|
|
static arma::mat probsLogistic(arma::mat const &src);
|
|
arma::mat toHidden(const arma::mat &v);
|
|
arma::mat toVisible(const arma::mat &h);
|
|
arma::mat uniform(size_t numRows, size_t numCols, double mu=0.0, double stdDev=1.0);
|
|
void uniform(arma::mat &srcDst, double mu=0.0, double stdDev=1.0);
|
|
private:
|
|
noise_gen_t m_noise;
|
|
arma::mat m_w;
|
|
arma::mat m_bh;
|
|
arma::mat m_bv;
|
|
};
|
|
|
|
#endif /* RBM_HPP */
|
|
|