symbolic regression python code