diff --git a/cskg/detectors/data_clumps.py b/cskg/detectors/data_clumps.py index adfad19..dd53726 100644 --- a/cskg/detectors/data_clumps.py +++ b/cskg/detectors/data_clumps.py @@ -123,9 +123,6 @@ def build_conditional_fp_tree(self): logger.debug(query) self.neo_db.cypher_query(query) - if len(results) > 100: - sleep(5) - # Query for frequent itemsets query = f""" MATCH path = (root:{ConditionalFpTreeNode.label})-[:LINKS*]->(end:{ConditionalFpTreeNode.label}) @@ -147,7 +144,11 @@ def build_conditional_fp_tree(self): if len(interim_nodes) > 1: self.result_collection.insert_one( - {"itemset": itemset, "support_count": frequency} + { + "itemset": itemset, + "support_count": frequency, + "size": len(itemset), + } ) # Clear CFP Tree