forked from stanfordnlp/stanza
-
Notifications
You must be signed in to change notification settings - Fork 0
/
coptic_cli.py
33 lines (29 loc) · 1.05 KB
/
coptic_cli.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import argparse
import sys
from os.path import join as j
from stanza.coptic import train, test, Predictor, _hyperparam_search
def main(args):
mode = args.mode
if mode == 'train':
train(
j("stanza/coptic_data", "scriptorium", "cs-ud-train-and-dev.conllu"),
j("stanza/coptic_data", "scriptorium", "cs-ud-minitest.conllu")
)
# train(
# j("coptic_data", "scriptorium", "cs-ud-train-preprocessed.conllu"),
# j("coptic_data", "scriptorium", "cs-ud-dev-preprocessed.conllu")
# )
elif mode == 'test':
test(j("stanza/coptic_data", "scriptorium", "cs-ud-test.conllu"))
elif mode == 'predict':
p = Predictor()
sys.stdout.buffer.write(p.predict(args.pred_file).encode("utf8"))
else:
_hyperparam_search()
if __name__ == '__main__':
ap = argparse.ArgumentParser()
ap.add_argument("mode", choices=['hyperopt', 'train', 'test', 'predict'])
ap.add_argument("--pred-file", type=str)
args = ap.parse_args()
sys.argv.pop()
main(args)