non linear regression python code