Skip to content
Snippets Groups Projects
Commit e59c75b3 authored by Maxime Petit's avatar Maxime Petit
Browse files

Fixed script and added print ts in main

parent d06eac84
No related branches found
No related tags found
No related merge requests found
...@@ -3,3 +3,4 @@ bin/* ...@@ -3,3 +3,4 @@ bin/*
.idea .idea
total_test_gold.conllu total_test_gold.conllu
total_test_predicted.conllu total_test_predicted.conllu
venv/*
\ No newline at end of file
...@@ -56,6 +56,8 @@ if __name__ == "__main__" : ...@@ -56,6 +56,8 @@ if __name__ == "__main__" :
transitionSet = [Transition(elem) for elem in (["SHIFT","REDUCE","LEFT","RIGHT"]+args.ts.split(',')) if len(elem) > 0] 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} strategy = {"RIGHT" : 1, "SHIFT" : 1, "LEFT" : 0, "REDUCE" : 0}
print("Transition Set :", [trans.name for trans in transitionSet])
if args.mode == "train" : 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) 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" : elif args.mode == "decode" :
......
...@@ -17,7 +17,7 @@ shift ...@@ -17,7 +17,7 @@ shift
for k in $(seq 0 $K) for k in $(seq 0 $K)
do do
>&2 echo "Training of "bin/$NAME"_$k :" >&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 "" >&2 echo ""
done done
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment