Word2vec coherence

gensim/corpora/__init__.py

@@ -12,6 +12,6 @@
 from .dictionary import Dictionary  # noqa:F401
 from .hashdictionary import HashDictionary  # noqa:F401
 from .wikicorpus import WikiCorpus  # noqa:F401
-from .textcorpus import TextCorpus  # noqa:F401
+from .textcorpus import TextCorpus, TextDirectoryCorpus  # noqa:F401
 from .ucicorpus import UciCorpus  # noqa:F401
 from .malletcorpus import MalletCorpus  # noqa:F401

gensim/models/basemodel.py

@@ -14,3 +14,11 @@ def print_topics(self, num_topics=20, num_words=10):
         probable words for `topics` number of topics to log.
         Set `topics=-1` to print all topics."""
         return self.show_topics(num_topics=num_topics, num_words=num_words, log=True)
+
+    def get_topics(self):
+        """
+        Returns:
+            np.ndarray: `num_topics` x `vocabulary_size` array of floats which represents
+            the term topic matrix learned during inference.
+        """
+        raise NotImplementedError