Spaces:
Running
Running
deepsource-autofix[bot]
commited on
Commit
•
23834a6
1
Parent(s):
9f82c1b
Format code with black
Browse files- pysr/sr.py +4 -2
pysr/sr.py
CHANGED
@@ -73,8 +73,8 @@ sympy_mappings = {
|
|
73 |
"div": lambda x, y: x / y,
|
74 |
"mult": lambda x, y: x * y,
|
75 |
"sqrt_abs": lambda x: sympy.sqrt(abs(x)),
|
76 |
-
"square": lambda x: x
|
77 |
-
"cube": lambda x: x
|
78 |
"plus": lambda x, y: x + y,
|
79 |
"sub": lambda x, y: x - y,
|
80 |
"neg": lambda x: -x,
|
@@ -341,6 +341,8 @@ def _add_sr_to_julia_project(Main, io_arg):
|
|
341 |
rev="v0.7.8",
|
342 |
)
|
343 |
Main.eval(f"Pkg.add(spec, {io_arg})")
|
|
|
|
|
344 |
class PySRRegressor(BaseEstimator, RegressorMixin):
|
345 |
def __init__(
|
346 |
self,
|
|
|
73 |
"div": lambda x, y: x / y,
|
74 |
"mult": lambda x, y: x * y,
|
75 |
"sqrt_abs": lambda x: sympy.sqrt(abs(x)),
|
76 |
+
"square": lambda x: x**2,
|
77 |
+
"cube": lambda x: x**3,
|
78 |
"plus": lambda x, y: x + y,
|
79 |
"sub": lambda x, y: x - y,
|
80 |
"neg": lambda x: -x,
|
|
|
341 |
rev="v0.7.8",
|
342 |
)
|
343 |
Main.eval(f"Pkg.add(spec, {io_arg})")
|
344 |
+
|
345 |
+
|
346 |
class PySRRegressor(BaseEstimator, RegressorMixin):
|
347 |
def __init__(
|
348 |
self,
|