commit 2fdc3642884a6283b806664e05356deb2f7c6890 Author: aboelhamd Date: Wed Apr 24 17:33:16 2019 +0200 Fix file closing in beamsearch and args number in merge-models diff --git a/merge-models.py b/merge-models.py index d6287fb..44cbbc2 100644 --- a/merge-models.py +++ b/merge-models.py @@ -2,7 +2,7 @@ from os import listdir from os.path import isfile, join import sys -if (len(sys.argv) != 4) : +if (len(sys.argv) != 3) : print('Usage: python merge-models.py modelsdest newfile'); sys.exit(-1) diff --git a/src/BeamSearch.cpp b/src/BeamSearch.cpp index 8d08969..8a396c6 100644 --- a/src/BeamSearch.cpp +++ b/src/BeamSearch.cpp @@ -164,20 +164,19 @@ main (int argc, char **argv) CLExec::beamSearch (&beamTree, beam, slTokens, newAmbigInfo, classesWeights, localeId); + // take the first sentence only + beamTree.erase (beamTree.begin () + 1, beamTree.end ()); + RuleExecution::getOuts (&outs, &combNodes, beamTree, nodesPool, ruleOutputs, spaces); // write the outs + for (unsigned j = 0; j < outs.size (); j++) + interInFile << outs[j] << endl; - if (interInFile.is_open ()) - { - for (unsigned j = 0; j < outs.size (); j++) - interInFile << outs[j] << endl; - } - - interInFile.close (); - lextorFile.close (); } + interInFile.close (); + lextorFile.close (); } else {