fix c_model.py

Author: Harry-Chan

w2v_classifier/ex_classifier/c_model.py

@@ -17,8 +17,8 @@ def __init__(self, input_dim, n_classes):
             nn.Tanh(),
             nn.Linear(self.hidden_dim, self.n_classes))
 
-    def forward(self, dataset):
-        sentence_vec, target = dataset
+    def forward(self, sentence_vec, target=None):
+        
         predicted = torch.tensor([[0.9, 0.1]], requires_grad=True)
         predicted_value, predicted_class = torch.max(predicted, 1)
 

w2v_classifier/ex_classifier/new_c_model.py

@@ -4,9 +4,7 @@
 
 from ex_classifier.a_sent2vec import *
 from ex_classifier.b_dataloader import SentDataloader
-# from ex_classifier.c_model import SentClassifier
-from ex_classifier.new_c_model import SentClassifier
-
+from ex_classifier.c_model import SentClassifier
 from ex_classifier.d_predict import sent_predictor