Создать более конкретный кластер MeanShift

Я использую инструмент SKleans MeanShift для кластеризации набора данных. Данные довольно близки друг к другу, и в настоящее время они группируются только в 2 группы. Я хотел бы добавить параметры в MeanShift, чтобы сделать его более точным. Я попытался установить пропускную способность в 1, но это вызвало проблему со списком вне диапазона. Вот мой текущий код:

ms = MeanShift(bandwidth=1.0)
ms.fit(c_o_c)
labels = ms.labels_
n_clusters_ = len(np.unique(labels))
cluster_centers = ms.cluster_centers_


for i in range(len(c_o_c)):
    plt.plot(c_o_c[i][0], c_o_c[i][1], colors[labels[i]], markersize = 1)

print(n_clusters_)
plt.scatter(cluster_centers[:,0], cluster_centers[:,1], marker="x", s=150, linewidths=5, zorder=10)
plt.show()

0 ответов

Другие вопросы по тегам