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

Corrected bug in decode

parent ade901ad
No related branches found
No related tags found
No related merge requests found
...@@ -113,7 +113,7 @@ if __name__ == "__main__" : ...@@ -113,7 +113,7 @@ if __name__ == "__main__" :
transNames = json.load(open(args.model+"/transitions.json", "r"))[1] transNames = json.load(open(args.model+"/transitions.json", "r"))[1]
args.predictedStr = transInfos[0] args.predictedStr = transInfos[0]
args.predicted = set({colName for colName in args.predictedStr.split(',')}) args.predicted = set({colName for colName in args.predictedStr.split(',')})
transitionSets = [[Transition(elem) for elem in transNames]for ts in transNames] transitionSets = [[Transition(elem) for elem in ts] for ts in transNames]
strategy = json.load(open(args.model+"/strategy.json", "r")) strategy = json.load(open(args.model+"/strategy.json", "r"))
printTS(transitionSets, sys.stderr) printTS(transitionSets, sys.stderr)
Decode.decodeMode(args.debug, args.corpus, args.type, transitionSets, strategy, args.reward, args.predicted, args.model) Decode.decodeMode(args.debug, args.corpus, args.type, transitionSets, strategy, args.reward, args.predicted, args.model)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment