diff --git a/source/poet.cpp b/source/poet.cpp index 0887884..335a1c0 100644 --- a/source/poet.cpp +++ b/source/poet.cpp @@ -138,7 +138,7 @@ arma::mat createTraining(const string &filename, size_t seq_len) } #define CREATE_TRAINING 0 -#define DO_TRAINING 1 +#define DO_TRAINING 0 #define DO_FORWARD 0 int main() @@ -210,16 +210,23 @@ int main() params.gibbsDoSampleVisible = false; stack->setParams(params); arma::mat state; - + arma::mat curr; for (int i=0; i < t_vc.n_rows; i++) { - arma::mat curr = stack->to_curr(t_vc.row(i)); + curr = stack->to_curr(t_vc.row(i)); arma::mat r = stack->step_forward(state, curr); arma::mat next = stack->to_next(r); char c = idx2ch(next.index_max()); putchar(c); } printf("\n"); + for (int i=0; i < 40; i++) + { + curr = stack->to_next(stack->step_forward(state, curr)); + char c = idx2ch(curr.index_max()); + putchar(c); + } + printf("\n"); printf("\nEnd of program\n"); return 0;