From 95bf6bfbd0df9fde02530be7e4f4f937242f4c33 Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Tue, 8 Nov 2016 22:53:57 -0600
Subject: [PATCH] extract word vectors

---
 .../extract_word_vecs.py                      | 41 +++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 answer_classifier_cached_features/extract_word_vecs.py

diff --git a/answer_classifier_cached_features/extract_word_vecs.py b/answer_classifier_cached_features/extract_word_vecs.py
new file mode 100644
index 0000000..177da2c
--- /dev/null
+++ b/answer_classifier_cached_features/extract_word_vecs.py
@@ -0,0 +1,41 @@
+from word2vec.word_vector_management import word_vector_manager
+import constants
+import ujson
+import numpy as np
+import os
+
+import tensorflow as tf
+
+model = '/home/tanmay/Code/GenVQA/Exp_Results/models_cvpr/' + \
+        'ans_through_obj_atr_rel_bin_feats/answer_classifiers/' + \
+        'model-68000'
+
+sess = tf.InteractiveSession()
+word_vec_mgr = word_vector_manager()
+model_restorer = tf.train.Saver()
+model_restorer.restore(sess, model)
+
+word2vec = word_vec_mgr.pretrained_word_vectors.eval()
+classifier_vecs = word_vec_mgr.word_vectors.eval()
+
+outdir = '/home/tanmay/Code/GenVQA/Exp_Results/models_cvpr/word_vectors'
+constants.mkdir_if_not_exists(outdir)
+
+# word2vec_file = os.path.join(outdir, 'word2vec.txt')
+# np.savetxt(
+#     word2vec_file,
+#     word2vec,
+#     delimiter=',')
+
+# classifier_vecs_file = os.path.join(outdir, 'classifier_vecs.txt')
+# np.savetxt(
+#     classifier_vecs_file,
+#     classifier_vecs,
+#     delimiter=',')
+    
+
+with open(constants.vocab_json) as file:
+    vocab = ujson.load(file)
+
+
+
-- 
GitLab