alteryx / alteryx/evalml

Large datasets consumes excessive amounts of memory when fitting ensemble pipelines

Open
#3,108 0 comments 1 reaction 1 assignee Claimed by @christopherbunn View on GitHub
performance
Dominant language
Python
Stars
850
Forks
96
PR merge metrics
No merged PRs in 30d

Description

When running nyc_taxi.csv dataset through AutoML search (which has ~1.5M rows and 19 columns), it uses up more than 30 GB of memory when fitting an ensembling pipeline. This excessive memory consumption causes AutoML search to crash, which prevents performance test runs from properly finishing.

This issue tracks finding the root cause of this memory consumption and potential fixes to mitigate this.

![image.png](https://images.zenhubusercontent.com/5d7a7c0116b8a20001cd5e9b/1fc20700-fdc0-42fd-816b-e3a9f90e3e9c)

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.