| 606 | random.seed() |
| 607 | |
| 608 | def test_lsa(self): |
| 609 | try: |
| 610 | import numpy |
| 611 | except ImportError, e: |
| 612 | print e |
| 613 | return |
| 614 | # Assert LSA properties. |
| 615 | k = 100 |
| 616 | lsa = vector.LSA(self.model, k) |
| 617 | self.assertEqual(lsa.model, self.model) |
| 618 | self.assertEqual(lsa.vectors, lsa.u) |
| 619 | self.assertEqual(set(lsa.terms), set(self.model.vector.keys())) |
| 620 | self.assertTrue(isinstance(lsa.u, dict)) |
| 621 | self.assertTrue(isinstance(lsa.sigma, list)) |
| 622 | self.assertTrue(isinstance(lsa.vt, list)) |
| 623 | self.assertTrue(len(lsa.u), len(self.model)) |
| 624 | self.assertTrue(len(lsa.sigma), len(self.model)-k) |
| 625 | self.assertTrue(len(lsa.vt), len(self.model)-k) |
| 626 | for document in self.model: |
| 627 | v = lsa.vectors[document.id] |
| 628 | self.assertTrue(isinstance(v, vector.Vector)) |
| 629 | self.assertTrue(len(v) == k) |
| 630 | print "pattern.vector.LSA" |
| 631 | |
| 632 | def test_lsa_concepts(self): |
| 633 | try: |