(self)
| 445 | print "pattern.vector.Model.frequent()" |
| 446 | |
| 447 | def test_cosine_similarity(self): |
| 448 | # Assert document cosine similarity. |
| 449 | v1 = self.model.similarity(self.model[0], self.model[1]) |
| 450 | v2 = self.model.similarity(self.model[0], self.model[2]) |
| 451 | v3 = self.model.similarity(self.model[0], vector.Document("cats cats")) |
| 452 | self.assertAlmostEqual(v1, 0.20, places=2) |
| 453 | self.assertAlmostEqual(v2, 0.00, places=2) |
| 454 | self.assertAlmostEqual(v3, 0.45, places=2) |
| 455 | # Assert that Model.similarity() is aware of LSA reduction. |
| 456 | try: |
| 457 | import numpy |
| 458 | self.model.reduce(2) |
| 459 | v1 = self.model.similarity(self.model[0], self.model[1]) |
| 460 | v2 = self.model.similarity(self.model[0], self.model[2]) |
| 461 | self.assertAlmostEqual(v1, 1.00, places=2) |
| 462 | self.assertAlmostEqual(v2, 0.00, places=2) |
| 463 | self.model.lsa = None |
| 464 | except ImportError, e: |
| 465 | pass |
| 466 | print "pattern.vector.Model.similarity()" |
| 467 | |
| 468 | def test_nearest_neighbors(self): |
| 469 | # Assert document nearest-neighbor search. |
nothing calls this directly
no test coverage detected