From baa176994e23bbb5ea5fed6a2a51e4e4145abf96 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 24 Oct 2019 15:10:23 +0200
Subject: [PATCH] Fixed attaching floating words in sentence when EOS is ref
and predicted
---
transition_machine/src/ActionBank.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/transition_machine/src/ActionBank.cpp b/transition_machine/src/ActionBank.cpp
index 6977f3a..753a663 100644
--- a/transition_machine/src/ActionBank.cpp
+++ b/transition_machine/src/ActionBank.cpp
@@ -769,10 +769,10 @@ std::vector<Action::BasicAction> ActionBank::str2sequence(const std::string & na
// Attaching floating words to new root
int sentenceEnd = b0;
auto & eos = c.getTape(ProgramParameters::sequenceDelimiterTape);
- while (sentenceEnd >= 0 && eos[sentenceEnd-b0] != ProgramParameters::sequenceDelimiter)
+ while (sentenceEnd >= 0 && eos.getHyp(sentenceEnd-b0) != ProgramParameters::sequenceDelimiter)
sentenceEnd--;
int sentenceStart = std::max(0,sentenceEnd-1);
- while (sentenceStart >= 0 && eos[sentenceStart-b0] != ProgramParameters::sequenceDelimiter)
+ while (sentenceStart >= 0 && eos.getHyp(sentenceStart-b0) != ProgramParameters::sequenceDelimiter)
sentenceStart--;
sentenceStart++;
--
GitLab