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

Add CustomFuncFit into problem; Add related examples

Browse Source
This commit is contained in:
root
2024-07-11 18:32:08 +08:00
parent 3cb5fbf581
commit be6a67d7e2
15 changed files with 241 additions and 437 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tensorneat/algorithm/hyperneat/substrate/default.py
View File

@@ -1,5 +1,7 @@
import jax.numpy as jnp
from . import BaseSubstrate
from jax import vmap, numpy as jnp
from .base import BaseSubstrate
from tensorneat.genome.utils import set_conn_attrs
class DefaultSubstrate(BaseSubstrate):
@@ -13,8 +15,9 @@ class DefaultSubstrate(BaseSubstrate):
def make_nodes(self, query_res):
return self.nodes
def make_conn(self, query_res):
return self.conns.at[:, 2:].set(query_res) # change weight
def make_conns(self, query_res):
# change weight of conns
return vmap(set_conn_attrs)(self.conns, query_res)
@property
def query_coors(self):