Merge pull request #112 from tomlincr/master

Add get_params method to BaseEstimator

karateclub/estimator.py

@@ -2,6 +2,7 @@
 import numpy as np
 import networkx as nx
 from typing import List
+import re
 
 """General Estimator base class."""
 
@@ -30,6 +31,13 @@ def get_memberships(self):
     def get_cluster_centers(self):
         """Getting the cluster centers."""
         pass
+
+    def get_params(self):
+        """Get parameter dictionary for this estimator.."""
+        rx = re.compile(r'^\_')
+        params = self.__dict__
+        params = {key: params[key] for key in params if not rx.search(key)}
+        return params
 
     def _set_seed(self):
         """Creating the initial random seed."""

test/test_base.py

@@ -0,0 +1,8 @@
+from karateclub import DeepWalk
+
+def test_get_params():
+    model = DeepWalk()
+    params = model.get_params()
+    assert len(params) != 0
+    assert type(params) is dict
+    assert '_embedding' not in params