-
-def affinity_clustering(similarities, output, damping=0.9, max_iter=100000, convergence_iter=30, preference_quantile=0.5, random_state=1968, verbose=True):
+from pathlib import Path
+from multiprocessing import cpu_count
+from dataclasses import dataclass
+from clustering_base import sim_to_dist, process_clustering_result, clustering_result, read_similarity_mat
+
+def affinity_clustering(similarities, output, *args, **kwargs):
+ subreddits, mat = read_similarity_mat(similarities)
+ clustering = _affinity_clustering(mat, *args, **kwargs)
+ cluster_data = process_clustering_result(clustering, subreddits)
+ cluster_data['algorithm'] = 'affinity'
+ return(cluster_data)
+
+def _affinity_clustering(mat, subreddits, output, damping=0.9, max_iter=100000, convergence_iter=30, preference_quantile=0.5, random_state=1968, verbose=True):