minor

lancopku · Aug 27, 2018 · 3f8f31d · 3f8f31d
1 parent 82f14ee
commit 3f8f31d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/model/auto_encoder.py b/model/auto_encoder.py
@@ -7,7 +7,7 @@
 
 class AutoEncoderModel(Seq2SeqModel):
     def __init__(self, name: str = 'auto'):
-        super(AutoEncoderModel, self).__init__(name)
+        super(AutoEncoderModel, self).__init__()
 
         self.q_dec = None
         self.q_target = None

diff --git a/model/base.py b/model/base.py
@@ -14,7 +14,7 @@
 
 class BasicChatbotModel(BasicModel):
     def __init__(self, name: str = 'basic_chatbot_model'):
-        super(BasicChatbotModel, self).__init__(name)
+        super(BasicChatbotModel, self).__init__()
 
         self.enc_length = None
         self.dec_length = None

diff --git a/model/seq.py b/model/seq.py
@@ -30,7 +30,7 @@ def __call__(self, x):
 
 class Seq2SeqModel(BasicChatbotModel):
     def __init__(self, name: str = 'seq'):
-        super(Seq2SeqModel, self).__init__(name)
+        super(Seq2SeqModel, self).__init__()
         self.q_enc = None
         self.a_dec = None
         self.a_target = None
@@ -166,10 +166,10 @@ def evaluate(self):
         self._restore_checkpoint()
 
         print('Start testing ...')
-        # batches = self._get_batches('test')
+        batches = self._get_batches('test')
         test_samples = [qa for qa, _ in self.dataset['test_samples']]
         test_samples_text = [text for _, text in self.dataset['test_samples']]
-        batches = [self._create_batch(test_samples)]
+        # batches = [self._create_batch(test_samples)]
         all_inputs = []
         all_outputs = []
         all_references = []
@@ -185,10 +185,10 @@ def evaluate(self):
             all_inputs += np.transpose(np.array(batch.q_enc_seq)).tolist()
             all_references += np.transpose(np.array(batch.a_target_seq)).tolist()
 
-        self._write_test_samples_literal(test_samples_text, all_outputs)
-        self._write_test_samples_results(all_outputs)
+        # self._write_test_samples_literal(test_samples_text, all_outputs)
+        # self._write_test_samples_results(all_outputs)
         # self._wirte_test_literal(all_inputs, all_outputs)
-        # self._write_evaluation_results(all_outputs, all_references)
+        self._write_evaluation_results(all_outputs, all_references)
 
     def _write_test_samples_results(self, outputs: List[List[int]]):
         results = {}