Skip to content

Commit de22762

Browse files
Added product aggregation function.
Removed unused imports.
1 parent d5ba24b commit de22762

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

neat/genome.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
from operator import mul
2+
from functools import reduce
3+
14
from neat.config import ConfigParameter, write_pretty_params
25
from neat.genes import DefaultConnectionGene, DefaultNodeGene
36
from neat.six_util import iteritems, itervalues, iterkeys
@@ -7,6 +10,9 @@
710

811
from random import choice, random, shuffle
912

13+
def product(x):
14+
return reduce(mul, x, 1.0)
15+
1016

1117
class DefaultGenomeConfig(object):
1218
__params = [ConfigParameter('num_inputs', int),
@@ -22,7 +28,7 @@ class DefaultGenomeConfig(object):
2228
ConfigParameter('node_delete_prob', float)]
2329

2430
allowed_connectivity = ['unconnected', 'fs_neat', 'full', 'partial']
25-
aggregation_function_defs = {'sum': sum, 'max': max, 'min': min}
31+
aggregation_function_defs = {'sum': sum, 'max': max, 'min': min, 'product':product}
2632

2733
def __init__(self, params):
2834
# Create full set of available activation functions.

neat/population.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
11
from __future__ import print_function
22

3-
import sys
4-
import time
5-
63
from neat.reporting import ReporterSet
74
from neat.species import SpeciesSet
85
from neat.six_util import iteritems, itervalues

0 commit comments

Comments
 (0)