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

@@ -5,7 +5,7 @@ from tensorneat.algorithm.neat import NEAT
from tensorneat.genome import DefaultGenome, BiasNode
from tensorneat.problem.rl import GymNaxEnv
from tensorneat.common import Act, Agg
from tensorneat.common import ACT, AGG
@@ -21,10 +21,10 @@ if __name__ == "__main__":
num_outputs=1,
init_hidden_layers=(),
node_gene=BiasNode(
activation_options=Act.tanh,
aggregation_options=Agg.sum,
activation_options=ACT.tanh,
aggregation_options=AGG.sum,
),
output_transform=Act.standard_tanh,
output_transform=ACT.standard_tanh,
),
),
problem=GymNaxEnv(