cavalab / srbench

A living benchmark framework for symbolic regression
https://cavalab.org/srbench/
GNU General Public License v3.0
203 stars 74 forks source link

dagsearch speedup #157

Closed kahlmeyer94 closed 10 months ago

kahlmeyer94 commented 10 months ago

It looks like the automatic testing failed because dagsearch took > 1h. Hence I lowered the number of iterations and added the possibility to use multiprocessing.

lacava commented 10 months ago

you can add parameters that are specific to testing if you want to just speed it up for the tests: here is an example from PySR's regressor.py of how to do it:

eval_kwargs = {
    "test_params": dict(
        niterations=3,
        ncyclesperiteration=500,
        populations=3,
    )
}

test_params is then set by calling est.set_params(**test_params) on your estimator

kahlmeyer94 commented 10 months ago

Okay thanks! Then I will rework my stuff and try again.