Update attmodel.py

goodnessSZW · Apr 17, 2018 · ccd075f · ccd075f
1 parent 0372b9e
commit ccd075f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/attmodel.py b/attmodel.py
@@ -66,6 +66,7 @@ def __init__(self, pad_len, num_rels, word_vectors, window_size, num_filters, em
             l2_loss += tf.nn.l2_loss(W)
             l2_loss += tf.nn.l2_loss(b)
 
+            # the implementation of Lin et al 2016 comes from https://github.com/thunlp/TensorFlow-NRE/blob/master/network.py
             sen_a = tf.get_variable("attention_A", [3*self.num_filters], initializer=tf.contrib.layers.xavier_initializer())
             sen_q = tf.get_variable("query", [3*self.num_filters, 1], initializer=tf.contrib.layers.xavier_initializer())
             sen_r = []