From b8ba4a7ada33cd1d9ee2c0af98458554c75ded48 Mon Sep 17 00:00:00 2001 From: John Bauer Date: Mon, 12 Sep 2022 19:16:17 -0700 Subject: Default trees written with format _O --- stanza/models/constituency/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/stanza/models/constituency/trainer.py b/stanza/models/constituency/trainer.py index 9a6ea4c1..8e270959 100644 --- a/stanza/models/constituency/trainer.py +++ b/stanza/models/constituency/trainer.py @@ -942,19 +942,19 @@ def run_dev_set(model, dev_trees, args, evaluator=None): else: with open(pred_file, 'w') as fout: for tree in treebank: - fout.write("{:_}".format(tree.predictions[0].tree)) + fout.write("{:_O}".format(tree.predictions[0].tree)) fout.write("\n") for i in range(args['num_generate']): pred_file = os.path.join(args['predict_dir'], args['predict_file'] + ".%03d.pred.mrg" % i) with open(pred_file, 'w') as fout: for tree in generated_treebanks[i+1]: - fout.write("{:_}".format(tree.predictions[0].tree)) + fout.write("{:_O}".format(tree.predictions[0].tree)) fout.write("\n") with open(orig_file, 'w') as fout: for tree in treebank: - fout.write("{:_}".format(tree.gold)) + fout.write("{:_O}".format(tree.gold)) fout.write("\n") if len(full_results) == 0: -- cgit v1.2.3