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

fix bugs

Browse Source
This commit is contained in:
root
2024-07-12 02:25:57 +08:00
parent 3194678a15
commit 5fc63fdaf1
28 changed files with 351 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/crossover_mutation.py
View File

@@ -1,5 +1,5 @@
import jax, jax.numpy as jnp
from tensorneat.common import Act
from tensorneat.common import ACT
from algorithm.neat import *
import numpy as np

6
test/nan_fitness.py
View File

@@ -1,5 +1,5 @@
import jax, jax.numpy as jnp
from tensorneat.common import Act
from tensorneat.common import ACT
from algorithm.neat import *
import numpy as np
@@ -17,8 +17,8 @@ def main():
max_nodes=20,
max_conns=20,
node_gene=DefaultNodeGene(
activation_options=(Act.tanh,),
activation_default=Act.tanh,
activation_options=(ACT.tanh,),
activation_default=ACT.tanh,
),
)

4
test/test_kan.ipynb
View File

@@ -27,7 +27,7 @@
"from algorithm.neat.gene.node.kan_node import KANNode\n",
"from algorithm.neat.gene.conn.bspline import BSplineConn\n",
"from problem.func_fit import XOR3d\n",
"from tensorneat.utils import Act\n",
"from tensorneat.utils import ACT\n",
"\n",
"import jax, jax.numpy as jnp\n",
"\n",
@@ -38,7 +38,7 @@
" max_conns=10,\n",
" node_gene=KANNode(),\n",
" conn_gene=BSplineConn(),\n",
" output_transform=Act.sigmoid, # the activation function for output node\n",
" output_transform=ACT.sigmoid, # the activation function for output node\n",
" mutation=DefaultMutation(\n",
" node_add=0.1,\n",
" conn_add=0.1,\n",

6
test/test_nan_fitness.py
View File

@@ -1,5 +1,5 @@
import jax, jax.numpy as jnp
from tensorneat.common import Act
from tensorneat.common import ACT
from algorithm.neat import *
import numpy as np
@@ -17,8 +17,8 @@ def main():
max_nodes=20,
max_conns=20,
node_gene=DefaultNodeGene(
activation_options=(Act.tanh,),
activation_default=Act.tanh,
activation_options=(ACT.tanh,),
activation_default=ACT.tanh,
),
)

4
test/test_update_by_batch.ipynb
View File

@@ -145,10 +145,10 @@
"source": [
"from algorithm.neat.gene.node.normalized import NormalizedNode\n",
"from algorithm.neat.gene.conn import DefaultConnGene\n",
"from tensorneat.utils import Act\n",
"from tensorneat.utils import ACT\n",
"\n",
"genome = DefaultGenome(num_inputs=3, num_outputs=2, max_nodes=10, max_conns=10,\n",
" node_gene=NormalizedNode(activation_default=Act.identity, activation_options=(Act.identity,)),\n",
" node_gene=NormalizedNode(activation_default=ACT.identity, activation_options=(ACT.identity,)),\n",
" conn_gene=DefaultConnGene(weight_init_mean=1))\n",
"state = genome.setup()\n",
"randkey = jax.random.key(0)\n",