Thesis/ADNC
1
0
Fork 0
mirror of https://github.com/JoergFranke/ADNC.git synced 2024-11-17 13:58:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

update mann

Browse Source
This commit is contained in:
Joerg Franke 2018-06-25 13:09:16 +02:00
parent 7a743e246d
commit 9f5b53486a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
adnc/model/mann.py
View File

@ -55,7 +55,10 @@ class MANN():
if self.input_embedding: if self.input_embedding:
word_idx_dict = self.input_embedding['word_idx_dict'] word_idx_dict = self.input_embedding['word_idx_dict']
embedding_size = self.input_embedding['embedding_size'] embedding_size = self.input_embedding['embedding_size']
tmp_dir = self.input_embedding['tmp_dir'] if 'tmp_dir' in self.input_embedding:
tmp_dir = self.input_embedding['tmp_dir']
else:
tmp_dir = "data_tmp"
glove = WordEmbedding(embedding_size, word_idx_dict=word_idx_dict, initialization='glove', tmp_dir=tmp_dir) glove = WordEmbedding(embedding_size, word_idx_dict=word_idx_dict, initialization='glove', tmp_dir=tmp_dir)
self._data = tf.placeholder(tf.int64, [None, self.batch_size], name='x') self._data = tf.placeholder(tf.int64, [None, self.batch_size], name='x')