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
Branches
No related tags found
No related merge requests found
...@@ -86,17 +86,16 @@ class Config : ...@@ -86,17 +86,16 @@ class Config :
def __len__(self) : def __len__(self) :
return len(self.lines) return len(self.lines)
# This print format is used by the script readTrace.py, avoid changes
def printForDebug(self, output) : def printForDebug(self, output) :
printedCols = ["ID","FORM","UPOS","HEAD","DEPREL"] printedCols = ["ID","FORM","UPOS","HEAD","DEPREL"]
left = 5 left = 5
right = 5 right = 5
historySize = 8
historyPopSize = 6
print("state :", self.state, file=output) 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("nbUndone :", self.nbUndone, file=output)
print("history :",[str(trans) for trans in self.history[-historySize:]], file=output) print("history :",[str(trans) for trans in self.history], file=output)
print("historyPop :",[(str(c[0]),"reward:"+str(c[3])) for c in self.historyPop[-historyPopSize:]], 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 = [] toPrint = []
for lineIndex in range(self.wordIndex-left, self.wordIndex+right) : for lineIndex in range(self.wordIndex-left, self.wordIndex+right) :
if lineIndex not in range(len(self.lines)) : if lineIndex not in range(len(self.lines)) :
......
...@@ -271,7 +271,6 @@ def trainModelRl(debug, networkName, modelDir, filename, nbIter, batchSize, devF ...@@ -271,7 +271,6 @@ def trainModelRl(debug, networkName, modelDir, filename, nbIter, batchSize, devF
probaOracle = list_probas[fromState][1] probaOracle = list_probas[fromState][1]
if debug : if debug :
print("-"*80, file=sys.stderr)
sentence.printForDebug(sys.stderr) sentence.printForDebug(sys.stderr)
action = selectAction(policy_net, state, transitionSet, sentence, missingLinks, probaRandom, probaOracle, fromState) 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