need to setup samplers

Author: Ashutosh Tiwari

datasets/__init__.py

@@ -0,0 +1,24 @@
+from typing import List, Tuple
+from torch_geometric.data import download_url
+
+
+class Dataset(object):
+    def __init__(self, root: str = '/tmp/', group_col: str = None, urls: List = None) -> None:
+        self.root = root
+        self.group_col = group_col
+        self.data = None
+        self.adj = None
+        self.file_paths = []
+        urls = urls or []
+        for url in urls:
+            self.file_paths.append(download_url(url, root))
+        self._set_data_df()
+        self._set_adj()
+
+    @property
+    def X(self):
+        return self.data.loc[:, self.data.columns != self.group_col].values
+
+    @property
+    def y(self):
+        return self.data[:, self.group_col].values

datasets/dataset.py

@@ -0,0 +1,18 @@
+import pandas as pd
+import networkx as nx
+
+
+from datasets import dataset
+
+
+class PolBlogs(dataset.Dataset):
+
+    def __init__(self, root: str = "/tmp/"):
+        super().__init__(root=root, group_col="political_leaning",
+                         urls=["https://websites.umich.edu/~mejn/netdata/polbooks.zip"])
+
+    def _set_data_df(self):
+        pass
+
+    def _set_adj(self):
+        pass