From e59c75b39a8555fc4d94d24f50cfc7aa3e684866 Mon Sep 17 00:00:00 2001
From: Maxime Petit <maxime.petit.3@etu.univ-amu.fr>
Date: Thu, 6 May 2021 14:22:40 +0200
Subject: [PATCH] Fixed script and added print ts in main
---
.gitignore | 1 +
main.py | 2 ++
trainKFolds.sh | 2 +-
3 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/.gitignore b/.gitignore
index fa2a13a..c42a158 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@ bin/*
.idea
total_test_gold.conllu
total_test_predicted.conllu
+venv/*
\ No newline at end of file
diff --git a/main.py b/main.py
index 026bce0..bcceaf3 100755
--- a/main.py
+++ b/main.py
@@ -56,6 +56,8 @@ if __name__ == "__main__" :
transitionSet = [Transition(elem) for elem in (["SHIFT","REDUCE","LEFT","RIGHT"]+args.ts.split(',')) if len(elem) > 0]
strategy = {"RIGHT" : 1, "SHIFT" : 1, "LEFT" : 0, "REDUCE" : 0}
+ print("Transition Set :", [trans.name for trans in transitionSet])
+
if args.mode == "train" :
Train.trainMode(args.debug, args.corpus, args.type, transitionSet, strategy, args.model, int(args.iter), int(args.batchSize), args.dev, args.bootstrap, args.silent)
elif args.mode == "decode" :
diff --git a/trainKFolds.sh b/trainKFolds.sh
index e008384..d1ba4fe 100755
--- a/trainKFolds.sh
+++ b/trainKFolds.sh
@@ -17,7 +17,7 @@ shift
for k in $(seq 0 $K)
do
>&2 echo "Training of "bin/$NAME"_$k :"
- ./main.py train $MODE data/UD_French-GSD_$k/train.conllu "bin/$NAME"_$k --dev data/UD_French-GSD_$k/dev.conllu $@
+ ./main.py train $MODE data/UD_French-GSD_$k/train.conllu "bin/$NAME"_$k --dev data/UD_French-GSD_$k/dev.conllu "$@"
>&2 echo ""
done
--
GitLab