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

add params key into setup.

Browse Source
This commit is contained in:
wls2002
2024-05-25 17:00:20 +08:00
parent bc378cec7f
commit 5626fddf41
5 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tensorneat/algorithm/neat/ga/crossover/base.py
View File

@@ -3,7 +3,7 @@ from utils import State
class BaseCrossover: class BaseCrossover:
def setup(self, state=State()): def setup(self, key, state=State()):
return state return state
def __call__(self, state, key, genome, nodes1, nodes2, conns1, conns2): def __call__(self, state, key, genome, nodes1, nodes2, conns1, conns2):

2
tensorneat/algorithm/neat/ga/mutation/base.py
View File

@@ -3,7 +3,7 @@ from utils import State
class BaseMutation: class BaseMutation:
def setup(self, state=State()): def setup(self, key, state=State()):
return state return state
def __call__(self, state, key, genome, nodes, conns, new_node_key): def __call__(self, state, key, genome, nodes, conns, new_node_key):

2
tensorneat/algorithm/neat/gene/base.py
View File

@@ -9,7 +9,7 @@ class BaseGene:
def __init__(self): def __init__(self):
pass pass
def setup(self, state=State()): def setup(self, key, state=State()):
return state return state
def new_attrs(self, state, key): def new_attrs(self, state, key):

2
tensorneat/algorithm/neat/genome/base.py
View File

@@ -24,7 +24,7 @@ class BaseGenome:
self.node_gene = node_gene self.node_gene = node_gene
self.conn_gene = conn_gene self.conn_gene = conn_gene
def setup(self, state=State()): def setup(self, key, state=State()):
return state return state
def transform(self, state, nodes, conns): def transform(self, state, nodes, conns):

7
tensorneat/algorithm/neat/species/base.py
View File

@@ -1,8 +1,9 @@
from utils import State from utils import State
class BaseSpecies: class BaseSpecies:
def setup(self, randkey): def setup(self, key, state=State()):
raise NotImplementedError return state
def ask(self, state: State): def ask(self, state: State):
raise NotImplementedError raise NotImplementedError
@@ -11,4 +12,4 @@ class BaseSpecies:
raise NotImplementedError raise NotImplementedError
def speciate(self, state, generation): def speciate(self, state, generation):
raise NotImplementedError raise NotImplementedError