diff --git a/reading_machine/src/Transition.cpp b/reading_machine/src/Transition.cpp index 879c074f8e8de9797097ba849194b8226e07d955..dd83119650e8a53f5198c00e854b18cb6505734d 100644 --- a/reading_machine/src/Transition.cpp +++ b/reading_machine/src/Transition.cpp @@ -101,8 +101,19 @@ void Transition::apply(Config & config, float entropy) { if (config.hasColIndex("ENTROPY")) { - auto action = Action::sumToHypothesis("ENTROPY", config.getWordIndex(), entropy); - action.apply(config, action); + if (name.find("SHIFT") == std::string::npos and name.find("REDUCE") == std::string::npos) + { + if (name.find("LEFT") != std::string::npos) + { + auto action = Action::sumToHypothesis("ENTROPY", config.getStack(0), entropy); + action.apply(config, action); + } + else + { + auto action = Action::sumToHypothesis("ENTROPY", config.getWordIndex(), entropy); + action.apply(config, action); + } + } } apply(config);