search

MilesCranmer / PySR

High-Performance Symbolic Regression in Python and Julia
https://astroautomata.com/PySR
Apache License 2.0
2.19k stars 207 forks source link

[Plot for Iterations/Generations with RMSE ] #254

Closed fitriaramlan closed 5 months ago

fitriaramlan commented 1 year ago

Do you have a feature to show a plot between iterations with RMSE? We have checked the dir(model), and we couldn't find it. If you don't have it, could you probably make that in the next pysr version?

MilesCranmer commented 1 year ago

Good idea. I don't have time to add it right now, but if you want to give it a shot, a PR would be welcome!

It would be easier to add this on the Julia side. Here is where the equations are printed to a file: https://github.com/MilesCranmer/SymbolicRegression.jl/blob/de0e51c89ae1a88cfc5a616a9d101b7a4d331dcb/src/SymbolicRegression.jl#L755-L766

you could probably add some code to append to a file over time, so that you can plot it, either live or at the end of training.

MilesCranmer commented 5 months ago

This will be added by https://github.com/MilesCranmer/SymbolicRegression.jl/pull/277. The new issue #581 is set up for tracking it, so let's move there.