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

initialize methods

Browse Source
This commit is contained in:
Priokin
2024-05-21 14:34:01 +08:00
parent 0e89ed1d7c
commit 40b7d8360c
46 changed files with 222 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tensorneat/algorithm/neat/gene/node/default.py
View File

@@ -60,6 +60,16 @@ class DefaultNodeGene(BaseNodeGene):
return jnp.array(
[self.bias_init_mean, self.response_init_mean, self.activation_default, self.aggregation_default]
)
def new_random_attrs(self, key):
return jnp.array([
mutate_float(key, self.bias_init_mean, self.bias_init_mean, self.bias_init_std,
self.bias_mutate_power, self.bias_mutate_rate, self.bias_replace_rate),
mutate_float(key, self.response_init_mean, self.response_init_mean, self.response_init_std,
self.response_mutate_power, self.response_mutate_rate, self.response_replace_rate),
self.activation_default,
self.aggregation_default,
])
def mutate(self, key, node):
k1, k2, k3, k4 = jax.random.split(key, num=4)