44 lines
1.3 KiB
Python
44 lines
1.3 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
|
|
)
|
|
|
|
def to_dict(self, state, conn):
|
|
in_idx, out_idx = conn[:2]
|
|
return {
|
|
"in": int(in_idx),
|
|
"out": int(out_idx),
|
|
}
|
|
|
|
def sympy_func(self, state, conn_dict, inputs):
|
|
raise NotImplementedError
|