def test_checkpoint(self): model1 = Model(checkpointDir=MODEL_CHECKPOINT_DIR) term = Term().createFromString("fox") for _ in range(5): prediction = model1.feedTerm(term) self.assertTrue(len(prediction.bitmap)) model1.save() model2 = Model(checkpointDir=MODEL_CHECKPOINT_DIR) model2.load() prediction = model2.feedTerm(term) self.assertTrue(len(prediction.bitmap))
parser.add_option( "-f", "--format", dest="format", help="Format to output (ie: csv)", metavar="FORMAT") parser.add_option( "-r", "--reset-sequences", dest="resetSequences", action="store_true", default=False, help="Reset the model sequence after every line") (options, args) = parser.parse_args() if not len(args): parser.print_help() print raise(Exception("file required")) model = Model(checkpointDir=options.checkpoint) try: readFile(args[0], model, resetSequences=options.resetSequences, format=options.format) except KeyboardInterrupt: if model.canCheckpoint(): print("Saving model before exiting...") model.save()
def testSaveWithoutCheckpointDirectory(self): model = Model() with self.assertRaises(Exception) as e: model.save() self.assertIn("No checkpoint directory specified", e.exception)