Regression problem with PyCaret - Codanics