python - 从黄砖 KElbowVisualizer 中提取 k

Question

我正在尝试从黄砖 KElbowVisualizer 可视化器中提取 k 的值以进行进一步处理。我可以在可视化上看到 k 值，但我似乎无法将其提取并放入变量中。

Answer 1

from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs

from yellowbrick.cluster import KElbowVisualizer

# Generate synthetic dataset with 8 random clusters
X, y = make_blobs(n_samples=1000, n_features=12, centers=8, random_state=42)

# Instantiate the clustering model and visualizer
model = KMeans()
visualizer = KElbowVisualizer(model, k=(4,12))

visualizer.fit(X)        # Fit the data to the visualizer
visualizer.show()        # Finalize and render the figure
visualizer.elbow_value_ # Get elbow value