add backend="jax" to sympy module

2024-06-13 05:55:33 +08:00
parent b3e442c688
commit 69d73aab73
12 changed files with 254 additions and 167 deletions
--- a/tensorneat/examples/interpret_visualize/genome_sympy.py
+++ b/tensorneat/examples/interpret_visualize/genome_sympy.py
@@ -28,3 +28,7 @@ if __name__ == '__main__':

    print(genome.repr(state, nodes, conns))
    print(network)
+
+    res = genome.sympy_func(state, network, precision=3)
+    print(res)
+