Using PySR, to perform integration and differentiation

[priya1809]

I'm interested in using PySR, to perform integration and differentiation. Has anyone successfully used for these purposes? I've been trying to use it, but I'm encountering some difficulties. Any guidance or advice would be greatly appreciated.

Code Error:

equation_results = []
for set_name, features in feature_sets.items():
X = merged_soil_df[features]
pysr_regressor = PySRRegressor(
    populations=40,
    niterations=10,
    binary_operators=["+", "-", "*", "/"],
    unary_operators=["sqrt", "log", "integrate"],
)
pysr_regressor.fit(X, y)
equations_df = pysr_regressor.equations_

C:\Users\ku03p\anaconda3\Lib\site-packages\pysr\sr.py:2766: UserWarning: Note: it looks like you are running in Jupyter. The progress bar will be turned off.
warnings.warn(
Traceback (most recent call last):

Cell In[248], line 10
pysr_regressor.fit(X, y)

File ~\anaconda3\Lib\site-packages\pysr\sr.py:2279 in fit
self._run(X, y, runtime_params, weights=weights, seed=seed, category=category)

File ~\anaconda3\Lib\site-packages\pysr\sr.py:1940 in _run
jl_op = jl.seval(op)

File ~.julia\packages\PythonCall\Nr75f\src\JlWrap\module.jl:27 in seval
return self._jl_callmethod($(pyjl_methodnum(pyjlmodule_seval)), expr)

JuliaError: UndefVarError: integrate not defined in Main
Stacktrace:
[1] top-level scope
@ :0
[2] eval
@ .\boot.jl:430 [inlined]
[3] eval
@ .\Base.jl:130 [inlined]
[4] pyjlmodule_seval(self::Module, expr::Py)
@ PythonCall.JlWrap C:\Users\ku03p.julia\packages\PythonCall\Nr75f\src\JlWrap\module.jl:13
[5] pyjl_callmethod(f::Any, self::Ptr{PythonCall.C.PyObject}, args_::Ptr{PythonCall.C.PyObject}, nargs::Int64)
@ PythonCall.JlWrap C:\Users\ku03p.julia\packages\PythonCall\Nr75f\src\JlWrap\base.jl:67
[6] _pyjl_callmethod(o::Ptr{PythonCall.C.PyObject}, args::Ptr{PythonCall.C.PyObject})
@ PythonCall.JlWrap.Cjl C:\Users\ku03p.julia\packages\PythonCall\Nr75f\src\JlWrap\C.jl:63

[MilesCranmer]

Hi @priya1809,

The operators available are listed on https://ai.damtp.cam.ac.uk/pysr/operators/

You can compute derivatives, but not as an operator, it has to be done in the context of a template expression spec: https://ai.damtp.cam.ac.uk/pysr/examples/#13-using-differential-operators

Cheers,
Miles