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

update a lot, take a break

Browse Source
This commit is contained in:
root
2024-07-12 07:47:33 +08:00
parent 58c56ab2ab
commit 99b8f7fd90
11 changed files with 2161 additions and 2418 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
examples/func_fit/xor.py
View File

@@ -1,5 +1,6 @@
from tensorneat.pipeline import Pipeline
from tensorneat import algorithm, genome, problem, common
from tensorneat import algorithm, genome, problem
from tensorneat.common import ACT
algorithm = algorithm.NEAT(
pop_size=10000,
@@ -8,7 +9,8 @@ algorithm = algorithm.NEAT(
genome=genome.DefaultGenome(
num_inputs=3,
num_outputs=1,
output_transform=common.ACT.sigmoid,
max_nodes=7,
output_transform=ACT.sigmoid,
),
)
problem = problem.XOR3d()
@@ -25,3 +27,20 @@ state = pipeline.setup()
state, best = pipeline.auto_run(state)
# show result
pipeline.show(state, best)
# visualize the best individual
network = algorithm.genome.network_dict(state, *best)
algorithm.genome.visualize(network, save_path="./imgs/xor_network.svg")
# transform the best individual to latex formula
from tensorneat.common.sympy_tools import to_latex_code, to_python_code
sympy_res = algorithm.genome.sympy_func(
state, network, sympy_output_transform=ACT.obtain_sympy(ACT.sigmoid)
)
latex_code = to_latex_code(*sympy_res)
print(latex_code)
# transform the best individual to python code
python_code = to_python_code(*sympy_res)
print(python_code)