beam collapse after every new sentence

decoder/src/Decoder.cpp

@@ -5,8 +5,6 @@
 #include "ProgramOutput.hpp"
 #include <chrono>
 
-//TODO : Faire se rejoindre les beam en debut de phrase, et faire la moyenne de totalentropy sur le nb d'actions de la phrase.
-
 Decoder::Decoder(TransitionMachine & tm, Config & config)
 : tm(tm), config(config)
 {
@@ -181,10 +179,20 @@ struct BeamNode
   TransitionMachine tm;
   Config config;
   Action * action;
+  int nbActions;
+
+
+  double getEntropy()
+  {
+    if (nbActions == 0)
+      return 0.0;
 
+    return totalEntropy / nbActions;
+  }
   BeamNode(TransitionMachine & tm, Config & config) : tm(tm), config(config)
   {
     totalEntropy = 0.0;
+    nbActions = 0;
     config.setOutputFile(nullptr);
     config.totalEntropy = totalEntropy;
   }
@@ -192,6 +200,7 @@ struct BeamNode
   {
     totalEntropy = other.totalEntropy + proba;
     this->action = action;
+    nbActions = other.nbActions + 1;
     config.setOutputFile(nullptr);
     config.totalEntropy = totalEntropy;
   }
@@ -285,7 +294,7 @@ void Decoder::decodeBeam()
   {
     std::sort(beam.begin(), beam.end(), [](std::shared_ptr<BeamNode> a, std::shared_ptr<BeamNode> b)
         {
-          return a->totalEntropy > b->totalEntropy;
+          return a->getEntropy() > b->getEntropy();
         });
   };
 
@@ -355,6 +364,17 @@ void Decoder::decodeBeam()
         computeAndPrintSequenceEntropy(config, justFlipped, errors, entropyAccumulator, nbActionsInSequence);
       }
     }
+
+    if (justFlipped)
+    {
+      beam.resize(std::min(1,(int)beam.size()));
+      if (!beam.empty())
+      {
+        beam[0]->totalEntropy = 0.0;
+        beam[0]->config.totalEntropy = 0.0;
+        beam[0]->nbActions = 0;
+      }
+    }
   }
 
   if (ProgramParameters::errorAnalysis)