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

- handle seq_lens without crash

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@851 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-20 18:14:35 +00:00
parent ce48f6f6c2
commit 9611c00213
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RnnTextHelper.cpp
+5 -1
View File
@@ -111,9 +111,13 @@ arma::mat RnnTextHelper::createTraining(const string& filename, size_t seq_len)
continue; continue;
} }
last_index = index; last_index = index;
pattern = arma::shift(pattern, 1);
arma::mat data = arma::zeros(1, NUM_CODES); arma::mat data = arma::zeros(1, NUM_CODES);
data[index] = 1; data[index] = 1;
if (seq_len > 1)
{
pattern = arma::shift(pattern, 1);
}
pattern.row(0) = data; pattern.row(0) = data;
arma::mat pattern_vector = pattern.as_row(); arma::mat pattern_vector = pattern.as_row();
batch.insert_rows(batch.n_rows, pattern_vector); batch.insert_rows(batch.n_rows, pattern_vector);