From ccd075f437f894ac1457b86433de8e4d9b6eca34 Mon Sep 17 00:00:00 2001 From: Tianyu Liu Date: Tue, 17 Apr 2018 11:01:48 +0800 Subject: [PATCH] Update attmodel.py --- attmodel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/attmodel.py b/attmodel.py index 26bd27d..e732050 100644 --- 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 = []