update baseline.py sentence_length

SouvickG · Dec 17, 2018 · a67fcb5 · a67fcb5
1 parent 47f84b8
commit a67fcb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baseline/baseline.py b/baseline/baseline.py
@@ -121,7 +121,7 @@ def get_text_model(self):
 		sentence_length = self.train_x.shape[2]
 
 		# Initializing sentence representation layers
-		embedding = Embedding(input_dim=self.vocabulary_size, output_dim=self.embedding_dim, weights=[self.data.W], input_length=self.sequence_length, trainable=False)
+		embedding = Embedding(input_dim=self.vocabulary_size, output_dim=self.embedding_dim, weights=[self.data.W], input_length=sentence_length, trainable=False)
 		conv_0 = Conv2D(self.num_filters, kernel_size=(self.filter_sizes[0], self.embedding_dim), padding='valid', kernel_initializer='normal', activation='relu')
 		conv_1 = Conv2D(self.num_filters, kernel_size=(self.filter_sizes[1], self.embedding_dim), padding='valid', kernel_initializer='normal', activation='relu')
 		conv_2 = Conv2D(self.num_filters, kernel_size=(self.filter_sizes[2], self.embedding_dim), padding='valid', kernel_initializer='normal', activation='relu')