我使用的k-means算法從scikit-learn庫,我想集群值在大熊貓數據幀3列:ID,value_1和value_2。我想使用value_1和來對信息進行聚類value_2,但是我也想ID與之保持關聯(因此我可以ID在每個聚類中創建的列表)。最好的方法是什么?目前,它也使用該ID數字進行聚類,這不是故意的。我當前的代碼(X是pandas數據框):kmeans = KMeans(n_clusters=2, n_init=3, max_iter=3000, random_state=1)(X_train, X_test) = train_test_split(X[['value_1','value_2']],test_size=0.30)kmeans = kmeans.fit(X_train)
添加回答
舉報
0/150
提交
取消
