Merge pull request #550 from LasyaPalla/main

Upadted getTransactions function in TransactionalDatabase.py
UdayLab · Nov 13, 2024 · ed7106d · ed7106d
2 parents 39c71cf + 8caf80a
commit ed7106d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py b/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py
@@ -156,10 +156,12 @@ def save(self, filename):
                 f.write(str(self.sep).join(map(str, line)) + '\n')
 
 
-    def getdataasDataframe(self,sep='\t'):
-        column = 'Transaction'
-        dataFrame = pd.DataFrame(colums=column)
-        dataFrame[column] = [sep.join(map(str,line) for line in self.data)]
+    def getTransactions(self,sep='\t'):
+        transactions = []
+        for line in self.data:
+            transactions.append(line)
+
+        dataFrame = pd.DataFrame([transactions], index=['TransactionItems']).T
         return dataFrame