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@853 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-20 18:53:10 +00:00
parent ae81da610b
commit 02b40561ec
3 changed files with 16 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RnnStack.hpp
+8 -19
View File
@@ -33,32 +33,21 @@ public:
arma::mat step_forward(arma::mat &state, const arma::mat &v);
arma::mat row_one_hot(size_t len, size_t index)
void clamp_one_hot(arma::mat &srcDst)
{
arma::mat data = arma::zeros(1, len);
data[index] = 1;
return data;
int index = srcDst.index_max();
srcDst = arma::zeros(arma::size(srcDst));
srcDst[index] = 1;
}
arma::mat to_next(const arma::mat &v, bool clamp=false)
arma::mat to_next(const arma::mat &v)
{
arma::mat res = v.cols(0, NUM_CODES-1);
if (clamp)
{
res = row_one_hot(res.n_cols, res.index_max());
}
return res;
return v.cols(0, NUM_CODES-1);
}
arma::mat to_curr(const arma::mat &v, bool clamp=false)
arma::mat to_curr(const arma::mat &v)
{
arma::mat res = v.cols(NUM_CODES, 2*NUM_CODES-1);
if (clamp)
{
res = row_one_hot(res.n_cols, res.index_max());
}
return res;
return v.cols(NUM_CODES, 2*NUM_CODES-1);
}
void setParams(Rbm::Params const &param)