search

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

Add arguments to change PCA biplot arrow and arrow label colors and other properties #1311

Open GegznaV opened 1 year ago

GegznaV commented 1 year ago

Describe the solution you'd like

In the PCA biplot, add a possibility to change the colors (and other properties) of arrows that represent features and their labels. This could be implemented via a dictionary-like argument.

Is your feature request related to a problem? Please describe. Currently, the PCA biplot arrow and their label colors are hardcoded in several places of method _draw_projection_features as "r" (red). And there is no possibility to pass other parameters to .plot() and .text() methods.

lwgray commented 1 year ago

This sounds like a great PR. Do you want to give it a shot?