def find_clusters(self, method, n_clusters=NUM_OF_CLUSTERS):
self.method = method
if len(self.snippets) == 0:
print(
"Sorry. There is nothing to cluster. Firstly, search for something."
)
return
if method.lower() == "stc":
self.clustering = SuffixTreeClustering(self.snippets)
elif method.lower() == "fca":
self.clustering = FCAClustering(self.snippets)
elif method.lower() == "ward":
self.clustering = HierarchicalClustering(self.snippets)
elif method.lower() == "k-means" or method.lower() == "kmeans":
self.clustering = kMeansClustering(self.snippets)
else:
print("Sorry, unknown clustering algorithm.")
return {}
self.clustering.find_clusters(n_clusters)
return self.get_clusters()
