tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update a lot, take a break

Browse Source
This commit is contained in:
root
2024-07-12 07:47:33 +08:00
parent 58c56ab2ab
commit 99b8f7fd90
11 changed files with 2161 additions and 2418 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/tensorneat/genome/base.py
View File

@@ -10,7 +10,7 @@ from tensorneat.common import (
StatefulBaseClass,
hash_array,
)
from .utils import valid_cnt
from .utils import valid_cnt, re_cound_idx
class BaseGenome(StatefulBaseClass):
@@ -160,7 +160,11 @@ class BaseGenome(StatefulBaseClass):
return nodes, conns
def network_dict(self, state, nodes, conns):
def network_dict(self, state, nodes, conns, whether_re_cound_idx=True):
if whether_re_cound_idx:
nodes, conns = re_cound_idx(
nodes, conns, self.get_input_idx(), self.get_output_idx()
)
return {
"nodes": self._get_node_dict(state, nodes),
"conns": self._get_conn_dict(state, conns),