dfc8f9198e
add ipynb test for testing whether add node or add conn will not change the output for the network.
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
import jax
|
|
from .. import BaseGene
|
|
|
|
|
|
class BaseConnGene(BaseGene):
|
|
"Base class for connection genes."
|
|
fixed_attrs = ["input_index", "output_index"]
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def new_zero_attrs(self, state):
|
|
# the attrs which make the least influence on the network, used in mutate add conn
|
|
raise NotImplementedError
|
|
|
|
def forward(self, state, attrs, inputs):
|
|
raise NotImplementedError
|
|
|
|
def update_by_batch(self, state, attrs, batch_inputs):
|
|
# default: do not update attrs, but to calculate batch_res
|
|
return (
|
|
jax.vmap(self.forward, in_axes=(None, None, 0))(state, attrs, batch_inputs),
|
|
attrs,
|
|
)
|
|
|
|
def repr(self, state, conn, precision=2, idx_width=3, func_width=8):
|
|
in_idx, out_idx = conn[:2]
|
|
in_idx = int(in_idx)
|
|
out_idx = int(out_idx)
|
|
|
|
return "{}(in: {:<{idx_width}}, out: {:<{idx_width}})".format(
|
|
self.__class__.__name__, in_idx, out_idx, idx_width=idx_width
|
|
)
|