Files
Castor/simple_qa_rnn/scripts/load_w2v.py
T
Salman Mohammed d12a9cb475 Relation prediction model from Simple QA paper (#20)
Initial implementation of RNNs for relation prediction described by Ture and Jojic:
https://arxiv.org/abs/1606.05029
2017-04-20 13:29:32 -04:00

11 lines
265 B
Python

import pickle
# load the pickled word vectors
w2v_pkl_path = "w2v_map_SQ.pkl"
print("loading word vectors from the pickle...")
w2v_map = None
with open(w2v_pkl_path, 'rb') as fh:
w2v_map = pickle.load(fh)
print(list(w2v_map.keys()))
print(len(w2v_map.keys()))