fix textCNN shaope comments typo

Guangzhan · May 24, 2020 · fc48317 · fc48317
1 parent cb4881e
commit fc48317
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/2-1.TextCNN/TextCNN-Torch.py b/2-1.TextCNN/TextCNN-Torch.py
@@ -48,7 +48,7 @@ def __init__(self):
         self.Bias = nn.Parameter(0.1 * torch.ones([num_classes])).type(dtype)
 
     def forward(self, X):
-        embedded_chars = self.W[X] # [batch_size, sequence_length, sequence_length]
+        embedded_chars = self.W[X] # [batch_size, sequence_length, embedding_size]
         embedded_chars = embedded_chars.unsqueeze(1) # add channel(=1) [batch, channel(=1), sequence_length, embedding_size]
 
         pooled_outputs = []
@@ -96,4 +96,4 @@ def forward(self, X):
 if predict[0][0] == 0:
     print(test_text,"is Bad Mean...")
 else:
-    print(test_text,"is Good Mean!!")
+    print(test_text,"is Good Mean!!")