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

View File

@@ -75,7 +75,7 @@ from pipeline import Pipeline
from algorithm.neat import *
from problem.rl_env import BraxEnv
from tensorneat.utils import Act
from tensorneat.utils import ACT
if __name__ == '__main__':
pipeline = Pipeline(
@@ -87,8 +87,8 @@ if __name__ == '__main__':
max_nodes=50,
max_conns=100,
node_gene=DefaultNodeGene(
activation_options=(Act.tanh,),
activation_default=Act.tanh,
activation_options=(ACT.tanh,),
activation_default=ACT.tanh,
)
),
pop_size=1000,