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

AttributeError: 'KMeans' object has no attribute 'k' #1267

Closed nipnipj closed 2 years ago

nipnipj commented 2 years ago

Example results in error:

from yellowbrick.cluster import KElbowVisualizer
from sklearn.cluster import KMeans
model = KElbowVisualizer(KMeans(), k=10)
model.fit(X)