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

clean imports and delete "create_XXX_functions"

Browse Source
This commit is contained in:
wls2002
2023-05-09 01:58:00 +08:00
parent f63a0c447b
commit 1f2327bbd6
7 changed files with 20 additions and 286 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
algorithms/neat/genome/genome.py
View File

@@ -22,27 +22,11 @@ from jax import numpy as jnp
from jax import jit
from jax import Array
from .activations import act_name2key
from .aggregations import agg_name2key
from .utils import fetch_first
EMPTY_NODE = np.array([np.nan, np.nan, np.nan, np.nan, np.nan])
def create_initialize_function(config):
pop_size = config.neat.population.pop_size
N = config.basic.init_maximum_nodes
num_inputs = config.basic.num_inputs
num_outputs = config.basic.num_outputs
default_bias = config.neat.gene.bias.init_mean
default_response = config.neat.gene.response.init_mean
default_act = act_name2key[config.neat.gene.activation.default]
default_agg = agg_name2key[config.neat.gene.aggregation.default]
default_weight = config.neat.gene.weight.init_mean
return partial(initialize_genomes, pop_size, N, num_inputs, num_outputs, default_bias, default_response,
default_act, default_agg, default_weight)
def initialize_genomes(pop_size: int,
N: int,
num_inputs: int,