How not to plot legend in RadViz plot? about yellowbrick HOT 4 CLOSED

I forgot to mention that I am using Python 3.9.0 and yellowbrick 1.5

from yellowbrick.

Hello @ivan-marroquin and thank you for using Yellowbrick!

In the Yellowbrick API, the show method calls an internal finalize method, which does things like adding a legend (you can see the RadViz version of finalize here). That means that even if you turn off the legend, if you later call show on the visualizer, the legend will still get added. To avoid the legend, you could do something like this:

import matplotlib.pyplot as plt
from yellowbrick.datasets import load_occupancy
from yellowbrick.features import RadViz

# Load the classification dataset
X, y = load_occupancy()

# Specify the target classes
classes = ["unoccupied", "occupied"]

# Instantiate the visualizer
visualizer = RadViz(classes=classes)

visualizer.fit(X, y) 
visualizer.transform(X) 
plt.show()

Thanks again for being a YB user!

from yellowbrick.

Hi @rebeccabilbro

Many thanks for your prompt answer. I checked my input data. Although there were no nan values, I found that there were duplicated values in some columns. So I added a bit of noise and then, I was able to generate the RadViz plot.

Hoping this tip can help other uses of this great package!

Ivan

from yellowbrick.

Ah, good catch @ivan-marroquin, and thanks so much for sharing your findings here for others to find!

from yellowbrick.