Fix Python lint issues

miyurud · May 17, 2024 · 7abe3f3 · 7abe3f3
1 parent 3dddd07
commit 7abe3f3
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src_python/models/supervised.py b/src_python/models/supervised.py
@@ -59,16 +59,28 @@ def initialize(self, **hyper_params):
         """
         if 'batch_size' not in hyper_params:
             batch_size = 20
+        else:
+            batch_size = hyper_params['batch_size']
         if 'layer_sizes' not in hyper_params:
-            num_samples = [20, 10]
-        if 'num_samples' not in hyper_params:
             layer_sizes = [10, 10]
+        else:
+            layer_sizes = hyper_params['layer_sizes']
+        if 'num_samples' not in hyper_params:
+            num_samples = [20, 10]
+        else:
+            num_samples = hyper_params['num_samples']
         if 'bias' not in hyper_params:
             bias = True
+        else:
+            bias = hyper_params['bias']
         if 'dropout' not in hyper_params:
             dropout = 0.1
+        else:
+            dropout = hyper_params['dropout']
         if 'lr' not in hyper_params:
             lr = 1e-2
+        else:
+            lr = hyper_params['lr']
 
         graph = sg.StellarGraph(nodes=self.nodes, edges=self.edges)