v0.0.2

Author: stephenleo

.gitignore

@@ -2,4 +2,5 @@
 __pycache__/
 build/
 dist/
-.DS_Store
+.DS_Store
+stripnet.egg-info/

README.md

@@ -64,4 +64,4 @@ STriP Net stands on the shoulder of giants and several prior work. The most nota
 If this work helped you in any way, please consider the following way to give me feedback so I can spend more time on this project
 1. ⭐ this repository
 2. ❤️ [the Huggingface space ](https://huggingface.co/spaces/stephenleo/strip)
-3. 📖 [Buy me a Book!](https://www.buymeacoffee.com/stephenleo)
+3. ☕ [Buy me a Coffee!](https://www.buymeacoffee.com/stephenleo)

notebook/test.ipynb

@@ -87,30 +87,30 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "2022-01-05 20:26:31 INFO: Load pretrained SentenceTransformer: allenai-specter\n",
-      "2022-01-05 20:26:49 INFO: Use pytorch device: cuda\n",
-      "2022-01-05 20:26:49 INFO: Missing data detected. Dropping them\n",
-      "2022-01-05 20:26:49 INFO: ========== Step1: Calculating Embeddings ==========\n",
-      "Batches: 100%|██████████| 3/3 [00:02<00:00,  1.43it/s]\n",
-      "2022-01-05 20:26:52 INFO: ========== Step2: Topic modeling ==========\n",
-      "2022-01-05 20:26:52 INFO: Initializing the topic model\n",
-      "2022-01-05 20:26:52 INFO: Training the topic model\n",
-      "2022-01-05 20:26:54,313 - BERTopic - Reduced dimensionality with UMAP\n",
-      "2022-01-05 20:26:54,324 - BERTopic - Clustered UMAP embeddings with HDBSCAN\n",
-      "2022-01-05 20:26:54 INFO: Populating Topic Results\n",
-      "2022-01-05 20:26:54 INFO: ========== Step3: STriP Network ==========\n",
-      "2022-01-05 20:26:54 INFO: Cosine similarity\n",
-      "2022-01-05 20:26:54 INFO: Calculating optimal threshold\n",
-      "2022-01-05 20:26:54 INFO: Number of connections: 126\n",
-      "2022-01-05 20:26:54 INFO: Calculating Network Plot\n",
-      "2022-01-05 20:26:54 INFO: ========== Model Fit Successfully! ==========\n"
+      "2022-01-05 20:42:12 INFO: Load pretrained SentenceTransformer: allenai-specter\n",
+      "2022-01-05 20:42:36 INFO: Use pytorch device: cuda\n",
+      "2022-01-05 20:42:36 INFO: Missing data detected. Dropping them\n",
+      "2022-01-05 20:42:36 INFO: ========== Step1: Calculating Embeddings ==========\n",
+      "Batches: 100%|██████████| 3/3 [00:02<00:00,  1.17it/s]\n",
+      "2022-01-05 20:42:41 INFO: ========== Step2: Topic modeling ==========\n",
+      "2022-01-05 20:42:41 INFO: Initializing the topic model\n",
+      "2022-01-05 20:42:41 INFO: Training the topic model\n",
+      "2022-01-05 20:42:50,425 - BERTopic - Reduced dimensionality with UMAP\n",
+      "2022-01-05 20:42:50,437 - BERTopic - Clustered UMAP embeddings with HDBSCAN\n",
+      "2022-01-05 20:42:50 INFO: Populating Topic Results\n",
+      "2022-01-05 20:42:50 INFO: ========== Step3: STriP Network ==========\n",
+      "2022-01-05 20:42:50 INFO: Cosine similarity\n",
+      "2022-01-05 20:42:50 INFO: Calculating optimal threshold\n",
+      "2022-01-05 20:42:50 INFO: Number of connections: 126\n",
+      "2022-01-05 20:42:50 INFO: Calculating Network Plot\n",
+      "2022-01-05 20:42:50 INFO: ========== Model Fit Successfully! ==========\n"
      ]
     },
     {
@@ -1141,6 +1141,35 @@
     "stripnet.most_important()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "['bertopic==0.9.4',\n",
+       " 'networkx==2.6.3',\n",
+       " 'numpy==1.22.0',\n",
+       " 'pandas==1.3.5',\n",
+       " 'plotly==5.5.0',\n",
+       " 'pyvis==0.1.9',\n",
+       " 'scikit_learn==1.0.2',\n",
+       " 'sentence_transformers==2.1.0',\n",
+       " 'setuptools==58.0.4']"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import pathlib\n",
+    "pathlib.Path(\"../requirements.txt\").read_text().splitlines()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

setup.py

@@ -4,11 +4,20 @@
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()
 
-required_packages = (pathlib.Path(__file__).parent / "requirements.txt").read_text().splitlines()
+# required_packages = (pathlib.Path(__file__).parent / "requirements.txt").read_text().splitlines()
+required_packages = ['bertopic==0.9.4',
+                     'networkx==2.6.3',
+                     'numpy==1.22.0',
+                     'pandas==1.3.5',
+                     'plotly==5.5.0',
+                     'pyvis==0.1.9',
+                     'scikit_learn==1.0.2',
+                     'sentence_transformers==2.1.0',
+                     'setuptools==58.0.4']
 
 setuptools.setup(
-    name="stripnet", 
-    version="0.0.1",
+    name="stripnet",
+    version="0.0.2",
     author="stephenleo",
     author_email="stephen.leo87@gmail.com",
     description="STriP Net: Semantic Similarity of Scientific Papers (S3P) Network",
@@ -27,4 +36,4 @@
         "Operating System :: OS Independent",
     ],
     python_requires='>=3.6, <3.9',
-)
+)