21 lines
490 B
Python
21 lines
490 B
Python
from neat.genome.aggregations import *
|
|
|
|
AGG_TOTAL_LIST = [sum_agg, product_agg, max_agg, min_agg, maxabs_agg, median_agg, mean_agg]
|
|
|
|
agg_name2key = {
|
|
'sum': 0,
|
|
'product': 1,
|
|
'max': 2,
|
|
'min': 3,
|
|
'maxabs': 4,
|
|
'median': 5,
|
|
'mean': 6,
|
|
}
|
|
|
|
|
|
def refactor_agg(config):
|
|
config['aggregation_default'] = agg_name2key[config['aggregation_default']]
|
|
config['aggregation_options'] = [
|
|
agg_name2key[act_name] for act_name in config['aggregation_options']
|
|
]
|