Python TopicModel.load Examples

Example #1

File: test_topic_model.py Project: jakemcc/textacy

 def test_save_load(self):
     filename = os.path.join(self.tempdir, 'model.pkl')
     expected = self.model.model.components_
     self.model.save(filename)
     tmp_model = TopicModel.load(filename)
     observed = tmp_model.model.components_
     self.assertEqual(observed.shape, expected.shape)
     self.assertTrue(np.equal(observed, expected).all())

Example #2

def test_save_load(tmpdir, model):
    filepath = str(tmpdir.join("model.pkl"))
    expected = model.model.components_
    model.save(filepath)
    tmp_model = TopicModel.load(filepath)
    observed = tmp_model.model.components_
    assert observed.shape == expected.shape
    assert np.equal(observed, expected).all()

Example #3

File: test_topic_model.py Project: EricSchles/textacy

 def test_save_load(self):
     filename = os.path.join(self.tempdir, 'model.pkl')
     expected = self.model.model.components_
     self.model.save(filename)
     tmp_model = TopicModel.load(filename)
     observed = tmp_model.model.components_
     self.assertEqual(observed.shape, expected.shape)
     self.assertTrue(np.equal(observed, expected).all())