Как изменить метки осей при использовании KElbowVisualizer от YellowBrick?

Question

Я использую следующий код для создания графика коэффициента силуэта с помощью KElbowVisualizer:

# Import the KElbowVisualizer method 

# Instantiate a scikit-learn K-Means model
model = KMeans(random_state=0)

# Instantiate the KElbowVisualizer with the number of clusters and the metric 
titleKElbow = "title"

visualizer = KElbowVisualizer(model, k=(2,7), metric='silhouette', timings=False,title = titleKElbow)

# Fit the data and visualize
visualizer.fit(df[['a','b','c']])    
visualizer.poof()  

На полученном графике метка оси x равна 'k'. Как я могу изменить метки осей на полученном графике? Я пробовал документацию , но, насколько мне известно, он показывает только, как добавлять метки осей в график в стиле plt.

Answer 1

Вы можете получить свойство ax визуализатора и напрямую использовать для него метод set_xlabel:

import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from yellowbrick.cluster import KElbowVisualizer


model = KMeans(random_state=0)
visualizer = KElbowVisualizer(
    model, 
    k=(2,7), 
    metric="silhouette", 
    timings=False,
    title="custom title"
)

visualizer.fit(df[["a", "b", "c"]])
visualizer.ax.set_xlabel("custom x label")
plt.show()

Спасибо за проверку Yellowbrick!