雑記： sklearn.cluster.KMeans で k-means 法 - クッキーの日記

sklearn.cluster.KMeans をつかってみたいときとかあると思います。iris データにつかうと以下のようになります。 # -*- coding: utf-8 -*- import pandas from sklearn import datasets from sklearn.cluster import KMeans from sklearn.metrics import confusion_matrix if __name__ == "__main__": iris = datasets.load_iris() predict = KMeans(n_clusters=3).fit(iris.data) mat = pandas.DataFrame(confusion_matrix(iris.target, predict.labels_)) mat.index = i