test_model.py 文件源码

def embedding_model_train_without_validation(self, model):
        history = model.train(self.texts, self.labels, epochs=2, batch_size=10, model_directory=self.model_directory,
                              verbose=0)
        self.assertIsInstance(history, History)
        self.assertTrue(os.path.exists(os.path.join(self.model_directory, "model.hd5")))
        self.assertTrue(os.path.exists(os.path.join(self.model_directory, "classifier.pk")))
        self.assertTrue(os.path.exists(os.path.join(self.model_directory, "description.txt")))
        self.assertTrue(os.path.exists(os.path.join(self.model_directory, "history.json")))