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

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",