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

fix bugs

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/gymnax/arcbot.py
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
@@ -24,8 +24,8 @@ if __name__ == "__main__":
num_outputs=3,
init_hidden_layers=(),
node_gene=BiasNode(
activation_options=Act.tanh,
aggregation_options=Agg.sum,
activation_options=ACT.tanh,
aggregation_options=AGG.sum,
),
output_transform=jnp.argmax,
),

6
examples/gymnax/cartpole.py
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
@@ -24,8 +24,8 @@ if __name__ == "__main__":
num_outputs=2,
init_hidden_layers=(),
node_gene=BiasNode(
activation_options=Act.tanh,
aggregation_options=Agg.sum,
activation_options=ACT.tanh,
aggregation_options=AGG.sum,
),
output_transform=jnp.argmax,
),

6
examples/gymnax/cartpole_hyperneat.py
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,
),

8
examples/gymnax/mountain_car_continuous.py
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(