ウォード法×ユークリッド距離によるクラスタリングを行うpythonプログラム

Question

ネット上から寄せ集めて下記のようなコードを作成しました。

from scipy.cluster.hierarchy import dendrogram, linkage, fcluster
import numpy as np

def hierarchical_clustering(emb, threshold):
    # 階層型クラスタリングの実施
    # ウォード法 x ユークリッド距離
    linkage_result = linkage(emb, method='ward', metric='euclidean')
    # クラスタ分けするしきい値を決める
    threshold_distance = threshold * np.max(linkage_result[:, 2])
    # クラスタリング結果の値を取得
    clustered = fcluster(linkage_result, threshold_distance, criterion='distance')
    print("end clustering.")
    return linkage_result, threshold_distance, clustered

def plot_dendrogram(linkage_result, doc_index, threshold):
    # 階層型クラスタリングの可視化
    plt.figure(facecolor='w', edgecolor='k')
    dendrogram(linkage_result, labels=doc_index, color_threshold=threshold)
    print("end plot.")
    plt.savefig('hierarchy.png')

def save_cluster(doc_index, clustered):
    doc_cluster = np.array([doc_index, clustered])
    doc_cluster = doc_cluster.T
    doc_cluster = doc_cluster.astype("int64")
    doc_cluster = doc_cluster[np.argsort(doc_cluster[:,1])]
    np.savetxt('cluster.csv', doc_cluster, delimiter=",", fmt="%.0f")
    print("save cluster.")

#事前に保存したDoc2Vec学習モデルを読み込み
m = models.Doc2Vec.load('test.model')

#ベクトルをリストに格納
vectors_list=[m.docvecs[n] for n in range(len(m.docvecs))]

#クラスタリング
threshold = 0.4
linkage_result, threshold, clustered = hierarchical_clustering(emb=vectors_list, threshold=threshold)
plot_dendrogram(linkage_result=linkage_result, doc_index=list(range(2000)), threshold=threshold)
save_cluster(list(range(2000)), clustered)

上記コードを実行した結果、下記のようなエラーが出力されました

---

ValueError Traceback (most recent call last)
<ipython-input-10-67000ab3edde> in <module>
39 plot_dendrogram(linkage_result=linkage_result, doc_index=list(range(2000)), threshold=threshold)
40 print(clustered)
---> 41 save_cluster(list(range(2000)), clustered)

<ipython-input-10-67000ab3edde> in save_cluster(doc_index, clustered)
23 doc_cluster = np.array([doc_index, clustered])
24 doc_cluster = doc_cluster.T
---> 25 doc_cluster = doc_cluster.astype("int64")
26 doc_cluster = doc_cluster[np.argsort(doc_cluster[:,1])]
27 np.savetxt('cluster.csv', doc_cluster, delimiter=",", fmt="%.0f")

ValueError: setting an array element with a sequence.

エラーの解消方法を教えてください

Comments

[meg_]

エラー発生行（doc_cluster = doc_cluster.astype("int64")）の前に print(doc_cluster) を入れると何が表示されますか？