请问如何对矩阵类型的 数据聚类？

import numpy as np
from sklearn.cluster import KMeans

x = np.array([[[1, 5, 6], [7, 4, 3]],
              [[9, 8, 4], [2, 4, 5]],
              [[7, 8, 5], [4, 1, 2]],
              [[8, 9, 5], [4, 1, 2]],
              [[1, 6, 6], [7, 4, 3]]])

km = KMeans(n_clusters=2, random_state=42).fit_predict(x)
print(km)

请问上面的代码应该怎么改才对

你这个数组是三维的啊，虽然不清楚sklearn的kmeans函数具体用法，但传入的X不是矩阵形式啊，应该是二维数组吧

`

先训练再预测嘛，指定个init吧

首先再审一次题，然后你私聊我