DistrictDataLabs / yellowbrick

Visual analysis and diagnostic tools to facilitate machine learning model selection.
http://www.scikit-yb.org/
Apache License 2.0
4.3k stars 559 forks source link

Fixes #1163 The inability to place FeatureCorrelation Visualizations into subplots. #1171

Closed lwgray closed 3 years ago

lwgray commented 3 years ago

When using ax param in the FeatureCorrelation visualizer to specify an axes location within a matplotlib subplot, it would not plot in the correct position. However, it would create an entirely new plot

To fix this, I changed the ax in the Super init so it was not explicitly setting it to None which was preventing axes from being specified

Before: image

After: image

Fixes #1163