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

optimize function reproduce for time usage

Browse Source
This commit is contained in:
wls2002
2023-05-08 16:28:13 +08:00
parent dde338696f
commit 8653f49826
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
algorithms/neat/pipeline.py
View File

@@ -2,7 +2,6 @@ from typing import List, Union, Tuple, Callable
import time
import jax
import jax.numpy as jnp
import numpy as np
from .species import SpeciesController
@@ -19,6 +18,7 @@ class Pipeline:
def __init__(self, config, seed=42):
self.generation_timestamp = time.time()
self.randkey = jax.random.PRNGKey(seed)
np.random.seed(seed)
self.config = config
self.N = config.basic.init_maximum_nodes
@@ -34,9 +34,6 @@ class Pipeline:
self.generation = 0
self.species_controller.init_speciate(self.pop_nodes, self.pop_connections)
# self.species_controller.speciate(self.pop_nodes, self.pop_connections,
# self.generation, self.o2o_distance, self.o2m_distance)
self.best_fitness = float('-inf')
def ask(self, batch: bool):