yellowbrick icon indicating copy to clipboard operation
yellowbrick copied to clipboard

Add more estimators to Silhouette Visualizer

Open gshigin opened this issue 4 years ago • 1 comments
trafficstars

At the moment Silhouette Visualizer can handle KMeans and MiniBatchKMeans. But as far as I know, the construction of silhouette plot isn't tied to centroid clustering in any way, but simply counts the score according to the specified metric. The only problem I can see here is that some algorithms, like OPTICS and DBSCAN, have a separate label for outliers.

Maybe I could help with this, however I'm new on GitHub, so I might need some help :)

gshigin avatar May 28 '21 12:05 gshigin

@gshigin it's a great idea and it would be great to see a PR for this!

bbengfort avatar Jul 10 '21 18:07 bbengfort

I came across the same problem, I think it might be as easy as replacing all self.estimator.n_clusters with len(np.unique(self.estimator.labels_)).

Is anyone working on this? Otherwise I'd give it a try

rpauli avatar Jan 16 '23 16:01 rpauli

@rpauli You are more than welcome to submit a PR for this.

lwgray avatar Jan 17 '23 20:01 lwgray