Я выполняю агломерационную кластеризацию. У меня есть модель, и я подгоняю к ней свой набор данных, но получаю ошибку при ее визуализации.
from sklearn.cluster import AgglomerativeClustering
cluster = AgglomerativeClustering(n_clusters=16, affinity='euclidean', linkage='ward')
cluster.fit_predict(small)
import matplotlib.pyplot as plt
%matplotlib inline
plt.figure(figsize=(10, 7))
plt.scatter(small['tracks'], small['genre_top'], c=cluster.labels_, cmap='rainbow')
Я получаю массив следующим образом:
и следующая ошибка: 