forked from KEMT/zpwiki
20 lines
710 B
Bash
20 lines
710 B
Bash
set -e
|
|
OUTDIR=build/train/output
|
|
TRAINDIR=build/train
|
|
mkdir -p $TRAINDIR
|
|
mkdir -p $OUTDIR
|
|
mkdir -p dist
|
|
# Delete old training results
|
|
rm -rf $OUTDIR/*
|
|
# Train dependency and POS
|
|
spacy train sk $OUTDIR ./build/input/slovak-treebank ./build/input/ud-artificial-gapping --n-iter 20 -p tagger,parser
|
|
rm -rf $TRAINDIR/posparser
|
|
mv $OUTDIR/model-best $TRAINDIR/posparser
|
|
# Train NER
|
|
# python ./train.py -t ./train.json -o $TRAINDIR/nerposparser -n 10 -m $TRAINDIR/posparser/
|
|
spacy train sk $TRAINDIR/nerposparser ./ner/train.json ./ner/eval.json --n-iter 20 -p ner
|
|
# Package model
|
|
spacy package $TRAINDIR/nerposparser dist --meta-path ./meta.json --force
|
|
cd dist/sk_sk1-0.2.0
|
|
python ./setup.py sdist --dist-dir ../
|