From 2f3c10095a8adca07ee3728c465c8c16455f1f31 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Wed, 19 Jan 2022 11:36:42 +0000 Subject: [PATCH] - update poet git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@843 b431acfa-c32f-4a4a-93f1-934dc6c82436 --- source/poet.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source/poet.cpp b/source/poet.cpp index 9efb219..ee06680 100644 --- a/source/poet.cpp +++ b/source/poet.cpp @@ -91,7 +91,7 @@ int main() arma::mat next; std::string curr_str; std::string next_str; - for (int i=0; i < t_vc.n_rows-50; i++) + for (int i=0; i < stack->getSeqLen(); i++) { curr = stack->to_curr(t_vc.row(i)); curr_str.append(1,RnnTextHelper::idx2ch(curr.index_max())); @@ -102,7 +102,7 @@ int main() cout << "Curr: " << curr_str << std::endl; cout << "Next: " << next_str << std::endl; - for (int i=0; i < 120; i++) + for (int i=stack->getSeqLen(); i < t_vc.n_rows; i++) { curr = next; curr_str.append(1,RnnTextHelper::idx2ch(curr.index_max()));