reorganized file structure

prediction-flow-sample.ipynb

@@ -503,6 +503,119 @@
     "\n",
     "pipe.transform(X=id_list)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import src.model\n",
+    "import importlib\n",
+    "\n",
+    "importlib.reload(src.model)\n",
+    "from src.model import main"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>title</th>\n",
+       "      <th>abstract</th>\n",
+       "      <th>id</th>\n",
+       "      <th>arxiv_subjects</th>\n",
+       "      <th>msc_tags</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>9786</th>\n",
+       "      <td>A logarithmic improvement in the two-point Wey...</td>\n",
+       "      <td>In this paper, we study the two-point Weyl Law...</td>\n",
+       "      <td>1905.05136v3</td>\n",
+       "      <td>[math.AP, math.SP]</td>\n",
+       "      <td>[35P20]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>39127</th>\n",
+       "      <td>Scaling Limit for the Kernel of the Spectral P...</td>\n",
+       "      <td>Let (M, g) be a compact smooth Riemannian mani...</td>\n",
+       "      <td>1411.0658v3</td>\n",
+       "      <td>[math.SP, math.AP, math.DG]</td>\n",
+       "      <td>None</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>46524</th>\n",
+       "      <td>On $L^p$-resolvent estimates and the density o...</td>\n",
+       "      <td>We address an interesting question raised by D...</td>\n",
+       "      <td>1204.3927v3</td>\n",
+       "      <td>[math.AP, math.CA]</td>\n",
+       "      <td>[58J50]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>38787</th>\n",
+       "      <td>C-infinity Scaling Asymptotics for the Spectra...</td>\n",
+       "      <td>This article concerns new off-diagonal estimat...</td>\n",
+       "      <td>1602.00730v1</td>\n",
+       "      <td>[math.AP, math-ph, math.DG, math.FA, math.MP, ...</td>\n",
+       "      <td>None</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>38299</th>\n",
+       "      <td>A lower bound for the $Θ$ function on manifold...</td>\n",
+       "      <td>In this short note, we prove that the usual $\\...</td>\n",
+       "      <td>1603.05697v1</td>\n",
+       "      <td>[math.DG, math.SP]</td>\n",
+       "      <td>None</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                                                   title  ... msc_tags\n",
+       "9786   A logarithmic improvement in the two-point Wey...  ...  [35P20]\n",
+       "39127  Scaling Limit for the Kernel of the Spectral P...  ...     None\n",
+       "46524  On $L^p$-resolvent estimates and the density o...  ...  [58J50]\n",
+       "38787  C-infinity Scaling Asymptotics for the Spectra...  ...     None\n",
+       "38299  A lower bound for the $Θ$ function on manifold...  ...     None\n",
+       "\n",
+       "[5 rows x 5 columns]"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "recs = main(id_list=[\"1905.05136v3\"])\n",
+    "\n",
+    "recs.head()"
+   ]
   }
  ],
  "metadata": {

src/model.py

@@ -6,10 +6,8 @@ from src.embedding import Embedder
 from src.search import Search
 
 
-def main():
-    id_list = [id]
-    path_to_library = ""
-    save_recs = False
+def main(id_list, save_recs=False):
+    path_to_library = "./data/libraries/APSP_50_allenai-specter"
     path_to_save_recs = ""
 
     ## Create pipeline
@@ -28,6 +26,8 @@ def main():
     if save_recs:
         recommendation_df.to_feather(path_to_save_recs)
 
+    return recommendation_df
+
 
 if __name__ == "main":
     main()