Skip to content
Snippets Groups Projects
Commit 6a6cd177 authored by Franck Dary's avatar Franck Dary
Browse files

Revert changes to config's debug print format that broke script readTrace

parent 423c26c3
No related branches found
No related tags found
No related merge requests found
......@@ -86,17 +86,16 @@ class Config :
def __len__(self) :
return len(self.lines)
# This print format is used by the script readTrace.py, avoid changes
def printForDebug(self, output) :
printedCols = ["ID","FORM","UPOS","HEAD","DEPREL"]
left = 5
right = 5
historySize = 8
historyPopSize = 6
print("state :", self.state, file=output)
print("stack :",[int(self.getAsFeature(ind, "ID")) for ind in self.stack], file=output)
print("stack :",[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[-historySize:]], file=output)
print("historyPop :",[(str(c[0]),"reward:"+str(c[3])) for c in self.historyPop[-historyPopSize:]], 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)
toPrint = []
for lineIndex in range(self.wordIndex-left, self.wordIndex+right) :
if lineIndex not in range(len(self.lines)) :
......
......@@ -271,7 +271,6 @@ 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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment