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))
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)
