I am running this relatively straightforward algorithm.
if I understand the algorithm correctly if you cluster to, say, 8 clusters, you should had the results for all clusters above 8, right?
Would you actually have to run the code multiple times, or how would you retrieve the intermediate clustering?
%%time
for k in K:
start_time = time.time() # Start timing
s[k] = []
db[k] = []
np.random.seed(123456) # for reproducibility
model = AgglomerativeClustering(linkage='ward', connectivity=w.sparse, n_clusters=k)
y = model.fit(cont_std)
cont_std_['AHC_k'+ str(k)] = y.labels_
silhouette_score = metrics.silhouette_score(cont_std, y.labels_, metric='euclidean')
print('silhouette at k=' + str(k) + ': ' + str(silhouette_score))
s[k].append(silhouette_score)
davies_bouldin_score = metrics.davies_bouldin_score(cont_std, y.labels_)
print(f'davies bouldin at k={k}: {davies_bouldin_score}')
db[k].append(davies_bouldin_score)
end_time = time.time() # End timing
print(f"Time for k={k}: {end_time - start_time} seconds") # Print the duration for the cycle
This is probably a rather roundabout way to get there, but it appears to work. I may yet try to clean this up later.
# Generate the list of nodes throughout the process,
# and an array that for each node index indicates the iteration
# at which it got merged with another.
nodes = [[i] for i in range(len(X))]
merged_at_stage = -np.ones(len(X) + len(model.children_), dtype=int)
for i, merge in enumerate(model.children_):
a, b = merge
nodes.append(nodes[a] + nodes[b])
merged_at_stage[a] = i
merged_at_stage[b] = i
# For a fixed number of clusters, identify the nodes
# at that point in the process
N_CLUSTERS = 2
clusters = [
nodes[i]
for i, x in enumerate(merged_at_stage)
if (
x >= len(X) - N_CLUSTERS # the node hasn't already been merged with another
and i <= len(X) + len(model.children_) - N_CLUSTERS # the node has already been created
)
]
clusters
is then a list of lists of indices. To turn that into a series of cluster labels (for scoring e.g.):
import pandas as pd
y_pred = pd.Series([-1] * len(X))
for i, cluster in enumerate(clusters):
y_pred[cluster] = i