MilesCranmer commited on
Commit
e968e20
·
1 Parent(s): f1c202a

Add new acosh/gamma functions

Browse files
Files changed (1) hide show
  1. pysr/sr.py +6 -5
pysr/sr.py CHANGED
@@ -39,7 +39,7 @@ sympy_mappings = {
39
  'acos': lambda x : sympy.acos(x),
40
  'asin': lambda x : sympy.asin(x),
41
  'atan': lambda x : sympy.atan(x),
42
- 'acosh':lambda x : sympy.acosh(x),
43
  'asinh':lambda x : sympy.asinh(x),
44
  'atanh':lambda x : sympy.atanh(x),
45
  'abs': lambda x : abs(x),
@@ -47,12 +47,13 @@ sympy_mappings = {
47
  'erf': lambda x : sympy.erf(x),
48
  'erfc': lambda x : sympy.erfc(x),
49
  'logm': lambda x : sympy.log(abs(x)),
50
- 'logm10':lambda x : sympy.log(abs(x), 10),
51
- 'logm2': lambda x : sympy.log(abs(x), 2),
52
- 'log1p': lambda x : sympy.log(x + 1),
53
- 'floor': lambda x : sympy.floor(x),
54
  'ceil': lambda x : sympy.ceil(x),
55
  'sign': lambda x : sympy.sign(x),
 
56
  }
57
 
58
  def pysr(X=None, y=None, weights=None,
 
39
  'acos': lambda x : sympy.acos(x),
40
  'asin': lambda x : sympy.asin(x),
41
  'atan': lambda x : sympy.atan(x),
42
+ 'acosh':lambda x : sympy.acosh(abs(x) + 1),
43
  'asinh':lambda x : sympy.asinh(x),
44
  'atanh':lambda x : sympy.atanh(x),
45
  'abs': lambda x : abs(x),
 
47
  'erf': lambda x : sympy.erf(x),
48
  'erfc': lambda x : sympy.erfc(x),
49
  'logm': lambda x : sympy.log(abs(x)),
50
+ 'logm10':lambda x : sympy.log(abs(x), 10),
51
+ 'logm2': lambda x : sympy.log(abs(x), 2),
52
+ 'log1p': lambda x : sympy.log(x + 1),
53
+ 'floor': lambda x : sympy.floor(x),
54
  'ceil': lambda x : sympy.ceil(x),
55
  'sign': lambda x : sympy.sign(x),
56
+ 'gamma': lambda x : sympy.gamma(x),
57
  }
58
 
59
  def pysr(X=None, y=None, weights=None,