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

debug-branch

Browse Source
This commit is contained in:
wls2002
2023-05-06 21:04:28 +08:00
parent 14fed83193
commit a85e6eba78
20 changed files with 1719 additions and 233 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
algorithms/neat/genome/graph.py
View File

@@ -95,11 +95,11 @@ def topological_sort_debug(nodes: Array, connections: Array) -> Array:
@vmap
def batch_topological_sort(nodes: Array, connections: Array) -> Array:
def batch_topological_sort(pop_nodes: Array, pop_connections: Array) -> Array:
"""
batch version of topological_sort
:param nodes:
:param connections:
:param pop_nodes:
:param pop_connections:
:return:
"""
return topological_sort(nodes, connections)
@@ -175,17 +175,17 @@ if __name__ == '__main__':
])
connections = jnp.array([
[
[0, 0, 1, 0, jnp.nan],
[0, 0, 1, 1, jnp.nan],
[0, 0, 0, 1, jnp.nan],
[0, 0, 0, 0, jnp.nan],
[jnp.nan, jnp.nan, 1, jnp.nan, jnp.nan],
[jnp.nan, jnp.nan, 1, 1, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, 1, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, jnp.nan, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, jnp.nan, jnp.nan]
],
[
[0, 0, 1, 0, jnp.nan],
[0, 0, 1, 1, jnp.nan],
[0, 0, 0, 1, jnp.nan],
[0, 0, 0, 0, jnp.nan],
[jnp.nan, jnp.nan, 1, jnp.nan, jnp.nan],
[jnp.nan, jnp.nan, 1, 1, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, 1, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, jnp.nan, jnp.nan],
[jnp.nan, jnp.nan, jnp.nan, jnp.nan, jnp.nan]
]
]