Files
tensorneat-mend/core/algorithm.py
2023-07-24 02:16:02 +08:00

29 lines
725 B
Python

from jax import Array
from .state import State
from .genome import Genome
EMPTY = lambda *args: args
class Algorithm:
def setup(self, randkey, state: State = State()):
"""initialize the state of the algorithm"""
pass
def ask(self, state: State):
"""require the population to be evaluated"""
pass
def tell(self, state: State, fitness):
"""update the state of the algorithm"""
pass
def forward(self, inputs: Array, transformed: Array):
"""the forward function of a single forward transformation"""
pass
def forward_transform(self, state: State, genome: Genome):
"""create the forward transformation of a genome"""
pass