From eebab89e2263a79186557807794845b1a7e39687 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Fri, 20 May 2022 13:03:28 +0200
Subject: [PATCH] Improved debug readability
---
Config.py | 8 +++++---
Train.py | 1 +
2 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/Config.py b/Config.py
index b26a965..e777680 100644
--- a/Config.py
+++ b/Config.py
@@ -90,11 +90,13 @@ class Config :
printedCols = ["ID","FORM","UPOS","HEAD","DEPREL"]
left = 5
right = 5
+ historySize = 8
+ historyPopSize = 6
print("state :", self.state, file=output)
- print("stack :",[self.getAsFeature(ind, "ID") for ind in self.stack], file=output)
+ print("stack :",[int(self.getAsFeature(ind, "ID")) for ind in self.stack], file=output)
print("nbUndone :", self.nbUndone, file=output)
- print("history :",[str(trans) for trans in self.history], file=output)
- print("historyPop :",[(str(c[0]),"dat:"+str(c[1]),"mvt:"+str(c[2]),"reward:"+str(c[3]),"state:"+str(c[4])) for c in self.historyPop], file=output)
+ print("history :",[str(trans) for trans in self.history[-historySize:]], file=output)
+ print("historyPop :",[(str(c[0]),"reward:"+str(c[3])) for c in self.historyPop[-historyPopSize:]], file=output)
toPrint = []
for lineIndex in range(self.wordIndex-left, self.wordIndex+right) :
if lineIndex not in range(len(self.lines)) :
diff --git a/Train.py b/Train.py
index ca19657..de8482e 100644
--- a/Train.py
+++ b/Train.py
@@ -271,6 +271,7 @@ def trainModelRl(debug, networkName, modelDir, filename, nbIter, batchSize, devF
probaOracle = list_probas[fromState][1]
if debug :
+ print("-"*80, file=sys.stderr)
sentence.printForDebug(sys.stderr)
action = selectAction(policy_net, state, transitionSet, sentence, missingLinks, probaRandom, probaOracle, fromState)
--
GitLab