Add CustomFuncFit into problem; Add related examples

2024-07-11 18:32:08 +08:00
parent 3cb5fbf581
commit be6a67d7e2
15 changed files with 241 additions and 437 deletions
--- a/tensorneat/common/activation/act_sympy.py
+++ b/tensorneat/common/activation/act_sympy.py
@@ -184,6 +184,12 @@ class SympyExp(sp.Function):
        return rf"\mathrm{{exp}}\left({sp.latex(self.args[0])}\right)"


+class SympySquare(sp.Function):
+    @classmethod
+    def eval(cls, z):
+        return sp.Pow(z, 2)
+
+
 class SympyAbs(sp.Function):
    @classmethod
    def eval(cls, z):