relevanceai.operations.cluster.algorithms.community_detection

Module Contents

class relevanceai.operations.cluster.algorithms.community_detection.CommunityDetection(threshold=0.75, min_community_size=10, init_max_size=1000, gpu=False)

fit_predict(self, vectors)

cosine(self, embeddings)

effecient cosine sim

topk(self, embeddings, k)

numpy topk

community_detection_cpu(self, embeddings)

community_detection_gpu(self, embeddings)