From 5c20b2432333c430feccb7b5a3ae9988631e0e9a Mon Sep 17 00:00:00 2001 From: Guan Hua Date: Mon, 30 May 2022 17:10:10 +0800 Subject: [PATCH] fix --- EGES_model_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/EGES_model_dataset.py b/EGES_model_dataset.py index d385cb4..0f16f26 100644 --- a/EGES_model_dataset.py +++ b/EGES_model_dataset.py @@ -50,7 +50,7 @@ def make_skipgram_loss(self): loss = tf.reduce_mean(tf.nn.sampled_softmax_loss( weights=self.softmax_w, biases=self.softmax_b, - labels=self.inputs[-1], + labels=self. batch_labels, inputs=self.merge_emb, num_sampled=self.n_samped, num_classes=self.num_nodes))