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

@@ -4,7 +4,7 @@ from tensorneat.pipeline import Pipeline
from tensorneat.algorithm.neat import NEAT
from tensorneat.algorithm.hyperneat import HyperNEAT, FullSubstrate
from tensorneat.genome import DefaultGenome
from tensorneat.common import Act
from tensorneat.common import ACT
from tensorneat.problem import GymNaxEnv
@@ -27,10 +27,10 @@ if __name__ == "__main__":
num_inputs=4, # size of query coors
num_outputs=1,
init_hidden_layers=(),
output_transform=Act.standard_tanh,
output_transform=ACT.standard_tanh,
),
),
activation=Act.tanh,
activation=ACT.tanh,
activate_time=10,
output_transform=jnp.argmax,
),