complete different samplers

Author: Ashutosh Tiwari

Makefile

@@ -41,3 +41,13 @@ run_tests:
 	@echo "running tests..."
 	@poetry install --only main --only test -vvv
 	@poetry run pytest -q tests
+
+.PHONY: run_notebook
+run_notebook:
+	@echo "running notebook..."
+	@poetry config virtualenvs.in-project true
+	@poetry install --only main --only debug -vvv
+	@poetry run poetry run jupyter contrib nbextension install --user
+	@poetry run jupyter nbextension enable --py codeium --user
+	@poetry run jupyter serverextension enable --py codeium --user
+	@poetry run jupyter notebook --ip=0.0.0.0 --port=8888 --no-browser

examples/sbm_sampler.ipynb

@@ -34,7 +34,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Using existing file polbooks.zip\n",
+      "Downloading https://websites.umich.edu/~mejn/netdata/polbooks.zip\n",
       "Extracting /tmp/polbooks.zip\n",
       "Using existing file polbooks.zip\n",
       "Extracting /tmp/polbooks.zip\n"
@@ -148,9 +148,53 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "id": "9c32f3b6",
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(3, 105)"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "block2node.toarray().shape"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "846a016e",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "array([[0.10477346, 0.60810079, 1.        ],\n",
+       "       [0.08979549, 0.8052493 , 1.        ],\n",
+       "       [0.07837984, 0.29670563, 1.        ]])"
+      ]
+     },
+     "execution_count": 10,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "block2block.toarray()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "881a4efc",
+   "metadata": {},
    "outputs": [],
    "source": []
   }

graph_ml/transformations/samplers.py

@@ -0,0 +1,17 @@
+# Author: Ashutosh Tiwari
+# Date: 3/19/24
+# Project: GraphML
+import numpy as np
+from scipy import sparse
+from ..utils import utils
+
+class Sampler(object):
+    """
+    Base class for all samplers
+    Every time there is a new adjacency matrix, sampler is required to be refitted.
+    May be should be changed in future, fit to be allowed to called multiple times.
+    """
+    def __init__(self):
+        pass
+
+    def fit(self, ):