added embdding model options and get documents topic method

ddangelov · Oct 15, 2020 · bc3ec52 · bc3ec52
1 parent 2b6939d
commit bc3ec52
Show file tree

Hide file tree

Showing 3 changed files with 510 additions and 157 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -29,7 +29,7 @@
 author = 'Dimo Angelov'
 
 # The full version, including alpha/beta/rc tags
-release = '1.0.11'
+release = '1.0.12'
 
 
 # -- General configuration ---------------------------------------------------

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 setuptools.setup(
     name="top2vec",
     packages=["top2vec"],
-    version="1.0.11",
+    version="1.0.12",
     author="Dimo Angelov",
     author_email="dimo.angelov@gmail.com",
     description="Top2Vec learns jointly embedded topic, document and word vectors.",
@@ -32,6 +32,11 @@
         'umap-learn',
         'hdbscan',
         'wordcloud',
+        'tensorflow',
+        'tensorflow_hub',
+        'tensorflow_text',
+        'torch',
+        'sentence_transformers',
       ],
     python_requires='>=3.6',
 )