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

- added void sample_one_hot()

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@856 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-21 09:52:53 +00:00
parent aef3049856
commit c2fe68a8dd
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RnnStack.hpp
+22
View File
@@ -15,6 +15,7 @@
#define RNNSTACK_HPP #define RNNSTACK_HPP
#include "AStack.hpp" #include "AStack.hpp"
#include "matutils.hpp"
class RnnStack : public AStack class RnnStack : public AStack
{ {
@@ -33,6 +34,27 @@ public:
arma::mat step_forward(arma::mat &state, const arma::mat &v); arma::mat step_forward(arma::mat &state, const arma::mat &v);
void sample_one_hot(arma::mat &srcDst)
{
double k = arma::accu(srcDst);
if (k != 0)
{
srcDst = srcDst / k;
}
arma::mat ps = Matutils::sample(srcDst);
if (arma::accu(ps) > 1)
{
arma::uvec q1 = find(ps > 0);
int winner = (q1.n_elem-1) * arma::randu(1)[0];
int wix = q1(winner);
srcDst = zeros(arma::size(srcDst));
srcDst[wix] = 1;
}
}
void clamp_one_hot(arma::mat &srcDst) void clamp_one_hot(arma::mat &srcDst)
{ {
int index = srcDst.index_max(); int index = srcDst.index_max();