jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- refactored

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@751 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-08 10:41:41 +00:00
parent 9472bb0a36
commit 637b237966
4 changed files with 39 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.hpp
+6 -5
View File
@@ -121,10 +121,6 @@ public:
void weightsInit(double stddev, double mu=0.0);
void train(arma::mat const &batch, IListener *pListener=nullptr);
arma::mat toHiddenState(const arma::mat &visible) const;
arma::mat toVisibleState(const arma::mat &hidden) const;
arma::mat toHiddenProbs(const arma::mat &visible) const;
arma::mat toVisibleProbs(const arma::mat &hidden) const;
static arma::mat normalize(const arma::mat &hidden);
const arma::mat& w() const;
const arma::mat& bv() const;
@@ -137,10 +133,15 @@ public:
{
return m_params;
}
static arma::mat prob(arma::mat const &src);
arma::mat v_to_h(const arma::mat &visible) const;
arma::mat h_to_v(const arma::mat &hidden) const;
arma::mat c_to_h(const arma::mat &context) const;
arma::mat h_to_c(const arma::mat &hidden) const;
private:
Params m_params;
arma::mat sample(arma::mat const &src);
static arma::mat probsLogistic(arma::mat const &src);
void weightUpdate(arma::mat const &v_state, arma::mat &dw, arma::mat &dbh, arma::mat &dbv);
void uniform(arma::mat &srcDst, double stdDev=1.0, double mu=0.5);