Improve initialization of spring layout k parameter

paulbrodersen · Dec 11, 2023 · 1c5cdae · 1c5cdae
1 parent 04fbce7
commit 1c5cdae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/netgraph/_node_layout.py b/netgraph/_node_layout.py
@@ -449,7 +449,8 @@ def get_fruchterman_reingold_layout(edges,
         fixed_node_sizes = np.array([])
 
     if k is None:
-        area = np.product(scale)
+        # area = np.product(scale)
+        area = np.pi * (min(scale) / 2) ** 2
         k = np.sqrt(area / float(total_nodes))
 
     temperatures = _get_temperature_decay(initial_temperature, total_iterations)