autogluon / autogluon/autogluon

Improve ease of use of `.plot_ensemble_model()`

Open
#2,079 2 comments 1 reaction 0 assignees View on GitHub
API & Doc module: tabular priority: 0
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.