deps updated

maximtrp · Jul 1, 2021 · b1d87e3 · b1d87e3
1 parent a04de59
commit b1d87e3
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -20,7 +20,7 @@ Refer to [documentation](https://bitermplus.readthedocs.io) to stay up to date.
 * pandas
 * scipy
 * scikit-learn
-* tmplot
+* tqdm
 
 ## Setup
 
@@ -85,7 +85,9 @@ perplexity = model.perplexity_
 coherence = model.coherence_
 
 # RESULTS VISUALIZATION
-btm.plot_model(model=model, docs=texts)
+# You need to install tmplot first
+import tmplot as tmp
+tmp.report(model=model, docs=texts)
 ```
 
 ![Report interface](images/topics_terms_plots.png)

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -1,10 +1,12 @@
 bitermplus
 ==========
 
-*Bitermplus* implements `Biterm topic model <https://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.402.4032&rep=rep1&type=pdf>`_
-for short texts introduced by Xiaohui Yan, Jiafeng Guo, Yanyan Lan, and Xueqi Cheng.
-Actually, it is a cythonized version of `BTM <https://github.com/xiaohuiyan/BTM>`_.
-This package is also capable of computing *perplexity* and *semantic coherence* metrics.
+*Bitermplus* implements `Biterm topic model
+<https://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.402.4032&rep=rep1&type=pdf>`_
+for short texts introduced by Xiaohui Yan, Jiafeng Guo, Yanyan Lan, and Xueqi
+Cheng. Actually, it is a cythonized version of `BTM
+<https://github.com/xiaohuiyan/BTM>`_. This package is also capable of computing
+*perplexity* and *semantic coherence* metrics.
 
 .. toctree::
    :maxdepth: 2
@@ -21,5 +23,4 @@ This package is also capable of computing *perplexity* and *semantic coherence*
 
    Model <bitermplus>
    Metrics <bitermplus.metrics>
-   Plotting functions <bitermplus.plot>
    Utility functions <bitermplus.util>
diff --git a/docs/source/install.rst b/docs/source/install.rst
@@ -37,4 +37,4 @@ Requirements
 * pandas
 * scipy
 * scikit-learn
-* tmplot
+* tqdm
diff --git a/setup.cfg b/setup.cfg
@@ -27,7 +27,6 @@ install_requires =
     pandas
     scipy
     scikit-learn
-    tmplot
     tqdm
 
 [options.packages.find]

diff --git a/src/bitermplus/_plot.py b/src/bitermplus/_plot.py
@@ -1,5 +1,5 @@
-__all__ = ['plot_model']
-from tmplot import report as plot_model
+#__all__ = ['plot_model']
+#from tmplot import report as plot_model
 
 # def vis_prepare_model(
 #         ttd: np.ndarray,