Here is how you can check if there is some bottleneck on your machine: Instead of running ./run.bat (or ./run.sh) you can run: python -m cProfile -o profile.pickle -m autogpt

But in 2.18, for the "regression" model type, I don't think the prediction variable ever hits _extract_output_and_other_columns and so it remains a dataframe.