0cb2f9473d
use jax.lax.while_loop in graph algorithms and forward function; fix "enabled not care" bug in forward
32 lines
745 B
Python
32 lines
745 B
Python
from functools import partial
|
|
|
|
import numpy as np
|
|
import jax
|
|
from jax import jit
|
|
|
|
from configs import Configer
|
|
from neat.pipeline import Pipeline
|
|
from neat.function_factory import FunctionFactory
|
|
|
|
xor_inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
|
|
xor_outputs = np.array([[0], [1], [1], [0]], dtype=np.float32)
|
|
|
|
def main():
|
|
config = Configer.load_config("xor.ini")
|
|
function_factory = FunctionFactory(config)
|
|
pipeline = Pipeline(config, function_factory)
|
|
forward_func = pipeline.ask()
|
|
# inputs = np.tile(xor_inputs, (150, 1, 1))
|
|
outputs = forward_func(xor_inputs)
|
|
print(outputs)
|
|
|
|
|
|
|
|
@jit
|
|
def f(x, jit_config):
|
|
return x + jit_config["bias_mutate_rate"]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|