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

update functions. Visualize, Interpretable and with evox

Browse Source
This commit is contained in:
root
2024-07-12 04:35:22 +08:00
parent 5fc63fdaf1
commit 0d6e7477bf
32 changed files with 207 additions and 427 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
examples/interpret_visualize/genome_sympy.py
View File

@@ -1,16 +1,16 @@
import jax, jax.numpy as jnp
from algorithm.neat import *
from algorithm.neat.genome.dense import DenseInitialize
from utils.graph import topological_sort_python
from tensorneat.genome import DefaultGenome
from tensorneat.common import *
from tensorneat.common.functions import SympySigmoid
if __name__ == "__main__":
genome = DenseInitialize(
genome = DefaultGenome(
num_inputs=3,
num_outputs=1,
max_nodes=50,
max_conns=500,
output_transform=ACT.sigmoid,
)
state = genome.setup()
@@ -22,7 +22,7 @@ if __name__ == "__main__":
input_idx, output_idx = genome.get_input_idx(), genome.get_output_idx()
res = genome.sympy_func(state, network, sympy_input_transform=lambda x: 999999999*x, sympy_output_transform=SympyStandardSigmoid)
res = genome.sympy_func(state, network, sympy_input_transform=lambda x: 999*x, sympy_output_transform=SympySigmoid)
(symbols,
args_symbols,
input_symbols,
@@ -35,3 +35,11 @@ if __name__ == "__main__":
inputs = jnp.zeros(3)
print(forward_func(inputs))
print(genome.forward(state, genome.transform(state, nodes, conns), inputs))
print(AGG.sympy_module("jax"))
print(AGG.sympy_module("numpy"))
print(ACT.sympy_module("jax"))
print(ACT.sympy_module("numpy"))