diff --git a/.gitignore b/.gitignore index fa2a13aabdd730b4290f238d663ece0c23eb2ed0..c42a158d8042b612560c1cc05459b15f5aaa6290 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 026bce0b9472bab5356b2e61deae02aaf60821b9..bcceaf374f1de756a876ba989f3942e885823b27 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 e008384456bbe10a542ac1f98af9f8024cb56037..d1ba4feedc231eb4b88fad942b0913717bfb648c 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