Python Model.load Examples

Example #1

File: model_api.py Project: rhyolight/nupic.fluent.server

def getModel(uid):
  if uid in modelCache:
    return modelCache[uid]

  modelDir = _getModelDir(uid)

  if not os.path.exists(modelDir):
    os.makedirs(modelDir)

  model = Model(checkpointDir=modelDir)

  if model.hasCheckpoint():
    model.load()

  modelCache[uid] = model
  return model

Example #2

File: model_test.py Project: chetan51/nupic.fluent

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

Example #3

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

Example #4

File: model_test.py Project: vsinha/nupic.fluent

  def testLoadWithoutCheckpointFile(self):
    model = Model(checkpointDir=MODEL_CHECKPOINT_DIR)

    with self.assertRaises(Exception) as e:
      model.load()
    self.assertIn("Could not find checkpoint file", e.exception)

Example #5

File: model_test.py Project: vsinha/nupic.fluent

  def testLoadWithoutCheckpointDirectory(self):
    model = Model()

    with self.assertRaises(Exception) as e:
      model.load()
    self.assertIn("No checkpoint directory specified", e.exception)