From ae202e7a60336d01101aaf5905a9fe3de3468cdf Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 25 Sep 2016 15:58:51 +0200 Subject: [PATCH] Fix init_model.py --- bin/init_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bin/init_model.py b/bin/init_model.py index a4a37d2ce..3d0a01542 100644 --- a/bin/init_model.py +++ b/bin/init_model.py @@ -148,7 +148,7 @@ def _read_senses(loc): return lexicon -def setup_vocab(get_lex_attr, tag_map, src_dir, dst_dir): +def setup_vocab(lex_attr_getters, tag_map, src_dir, dst_dir): if not dst_dir.exists(): dst_dir.mkdir() @@ -157,7 +157,7 @@ def setup_vocab(get_lex_attr, tag_map, src_dir, dst_dir): write_binary_vectors(vectors_src.as_posix, (dst_dir / 'vec.bin').as_posix()) else: print("Warning: Word vectors file not found") - vocab = Vocab(get_lex_attr=get_lex_attr, tag_map=tag_map) + vocab = Vocab(lex_attr_getters=lex_attr_getters, tag_map=tag_map) clusters = _read_clusters(src_dir / 'clusters.txt') probs, oov_prob = _read_probs(src_dir / 'words.sgt.prob') if not probs: