autogluon / autogluon/autogluon
Improve ease of use of `.plot_ensemble_model()`
- Dominant language
- Python
- Stars
- 10.7k
- Forks
- 1.2k
- Avg merge
- 21h 29m
- Merged PRs (30d)
- 57
Description
### Describe the issue linked to the documentation
Due to dependency requirements, `predictor.plot_ensemble_model()` is not easy to use.
https://github.com/awslabs/autogluon/blob/master/tabular/src/autogluon/tabular/predictor/predictor.py#L2659
### Suggest a potential alternative/fix
Refactor `predictor.plot_ensemble_model()` and make the dependency rely on a package that is installed by default in AutoGluon and works across Linux, Mac, and Windows. The refactor should also work in notebook environments.
Contributor guide
Research direction
Start at tabular/src/autogluon/tabular/predictor/predictor.py around line 2659 and inspect predictor.plot_ensemble_model(). Check how its current dependency is used, then verify the refactor with Linux, macOS, Windows, and notebook environments. Done means the method relies on a package installed by default and is easy to use across those environments.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Refactor
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100