From bf5a7cc598964f90d4f2e50ef915b87e22e27f87 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 2 Feb 2016 22:30:00 +0100 Subject: [PATCH] * Update train_pos_tagger example --- examples/train_pos_tagger.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/train_pos_tagger.py b/examples/train_pos_tagger.py index b1750ea64..43bd607c7 100644 --- a/examples/train_pos_tagger.py +++ b/examples/train_pos_tagger.py @@ -62,8 +62,10 @@ def main(output_dir): tokens = tokenizer.tokens_from_list(words) tagger.train(tokens, tags) random.shuffle(DATA) - tagger.model.end_training(path.join(output_dir, 'pos', 'model')) - vocab.strings.dump(path.join(output_dir, 'vocab', 'strings.txt')) + tagger.model.end_training() + tagger.model.dump(path.join(output_dir, 'pos', 'model')) + with io.open(output_dir, 'vocab', 'strings.json') as file_: + tagger.vocab.strings.dump(file_) if __name__ == '__main__':