add save and load function for classes.

2024-06-09 20:33:02 +08:00
parent 374c05f5b7
commit 52e5d603f5
13 changed files with 70 additions and 42 deletions
--- a/tensorneat/algorithm/neat/ga/crossover/base.py
+++ b/tensorneat/algorithm/neat/ga/crossover/base.py
@@ -1,9 +1,6 @@
-from utils import State
+from utils import StatefulBaseClass


-class BaseCrossover:
-    def setup(self, state=State()):
-        return state
-
+class BaseCrossover(StatefulBaseClass):
    def __call__(self, state, randkey, genome, nodes1, nodes2, conns1, conns2):
        raise NotImplementedError