From 18c40b862cdc342378086ce9e91c2029644c1577 Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Wed, 12 Oct 2016 09:17:59 +0100 Subject: [PATCH] Index confliction It currently does not have a problem. --- model/sswe.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/model/sswe.py b/model/sswe.py index 65a8ed4..13b0438 100644 --- a/model/sswe.py +++ b/model/sswe.py @@ -233,10 +233,10 @@ def train(self, correct_sequences, y_score): dnoise_inputs = dnoise_inputss[index].reshape(( self.window_size, self.embedding_size)) - for (i, di) in zip(correct_sequence, dcorrect_inputs): - self.embeddings[i] -= self.embedding_learning_rate * di - for (i, di) in zip(noise_sequence, dnoise_inputs): - self.embeddings[i] -= self.embedding_learning_rate * di + for (e_i, di) in zip(correct_sequence, dcorrect_inputs): + self.embeddings[e_i] -= self.embedding_learning_rate * di + for (e_i, di) in zip(noise_sequence, dnoise_inputs): + self.embeddings[e_i] -= self.embedding_learning_rate * di self.train_cnt += len(correct_sequences)