from pcfg import PCFG import argparse parser = argparse.ArgumentParser() parser.add_argument("--corpus", help="training treebank corpus", type=str) parser.add_argument("--sentences", help="raw token sentences", type=str) parser.add_argument("--outfile", help="name of the output file", type=str) args = parser.parse_args() grammar = PCFG(args.corpus) grammar.parse_corpus() grammar.predict(args.sentences, args.outfile)