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

- bring back calcContextBatch

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@840 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-19 11:01:37 +00:00
parent 22833df1d8
commit 963ee3cd9d
3 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Layer.hpp
+2 -1
View File
@@ -48,7 +48,8 @@ public:
bool weightsLoad(std::string const &dir, std::string const &prj);
bool weightsSave(std::string const &dir, std::string const &prj);
void train(arma::mat const &batch, IListener *pListener=nullptr);
void calcContextBatch(arma::mat &batch);
void train(arma::mat const &batch, IListener *pListener=nullptr);
arma::mat to_h_gibbs(const arma::mat& v_probs);
arma::mat to_v_gibbs(const arma::mat& h_probs);