Merge branch 'UdayLab:main' into main

Author: s1312007-Ojash

PAMI/extras/graph/plotFrequentPatternSetsGraph.py

@@ -27,6 +27,8 @@
      along with this program.  If not, see <https://www.gnu.org/licenses/>.
 """
 
+import sys
+
 import matplotlib.pyplot as plt
 from typing import Dict, Union
 
@@ -78,31 +80,23 @@ def save(self, oFile: str) -> None:
         plt.savefig(oFile)
         plt.close()
 
-    def getStatistics(self) -> Dict[str, Union[int, float]]:
+    def getStatistics(self) -> None:
         if not self.patternCounts:
-            return {}
+            return None
+
+        minLen = min(self.patternCounts.keys())
+        maxLen = max(self.patternCounts.keys())
 
-        return {
-            'totalPatterns': sum(self.patternCounts.values()),
-            'minPatternSize': min(self.patternCounts.keys()),
-            'maxPatternSize': max(self.patternCounts.keys()),
-            'patternSizeDistribution': self.patternCounts
-        }
+        print("Statistics:")
+        print(f" Length range: {minLen}-{maxLen}")
+        print(" Pattern Size Distribution <size: #count>:")
+        for size, count in sorted(self.patternCounts.items()):
+            print(f"   {size}:{count}")
+        return None
 
 
 if __name__ == "__main__":
-    samplePatterns = {
-        ('A',): 100,
-        ('B',): 150,
-        ('C',): 120,
-        ('A', 'B'): 80,
-        ('A', 'C'): 70,
-        ('B', 'C'): 90,
-        ('A', 'B', 'C'): 50,
-        ('D',): 110,
-        ('A', 'D'): 60,
-        ('B', 'D'): 65,
-    }
-
-    obj = plotFrequentPatternSetsGraph(samplePatterns)
+    # Usage: python plotFrequentPatternSetsGraph.py patterns.txt
+    obj = plotFrequentPatternSetsGraph(sys.argv[1])
     obj.plot()
+    obj.getStatistics()