From 8cc02de52bb0534c350df54cf2360310dabc8e82 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 6 May 2021 18:00:42 +0200
Subject: [PATCH] Print TransitionSet in main
---
main.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/main.py b/main.py
index a618b92..9509334 100755
--- a/main.py
+++ b/main.py
@@ -62,11 +62,13 @@ if __name__ == "__main__" :
if args.mode == "train" :
json.dump([t.name for t in transitionSet], open(args.model+"/transitions.json", "w"))
json.dump(strategy, open(args.model+"/strategy.json", "w"))
+ print("Transition Set :", [trans.name for trans in transitionSet], file=sys.stderr)
Train.trainMode(args.debug, args.corpus, args.type, transitionSet, strategy, args.model, int(args.iter), int(args.batchSize), args.dev, args.bootstrap, args.incr, args.silent)
elif args.mode == "decode" :
transNames = json.load(open(args.model+"/transitions.json", "r"))
transitionSet = [Transition(elem) for elem in transNames]
strategy = json.load(open(args.model+"/strategy.json", "r"))
+ print("Transition Set :", [trans.name for trans in transitionSet], file=sys.stderr)
Decode.decodeMode(args.debug, args.corpus, args.type, transitionSet, strategy, args.model)
else :
print("ERROR : unknown mode '%s'"%args.mode, file=sys.stderr)
--
GitLab