Remove printing for useless nodes when generating python code or latex formula

2025-04-16 10:43:38 +08:00
parent 5fa5e81c72
commit ffc1f09be0
4 changed files with 39 additions and 5 deletions
--- a/src/tensorneat/genome/utils.py
+++ b/src/tensorneat/genome/utils.py
@@ -98,7 +98,7 @@ def re_cound_idx(nodes, conns, input_idx, output_idx):
    for i, key in enumerate(nodes[:, 0]):
        if np.isnan(key):
            continue
-        if np.in1d(key, input_idx + output_idx):
+        if np.isin(key, input_idx + output_idx):
            continue
        old2new[int(key)] = next_key
        next_key += 1