Update data_clumps.py

yunfan-yang · Apr 7, 2024 · 3a6e8b6 · 3a6e8b6
1 parent 7d45222
commit 3a6e8b6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git 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