Update TransactionalDatabase.py

UdayLab · May 7, 2024 · e4646c9 · e4646c9
1 parent 072c662
commit e4646c9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py b/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py
@@ -208,8 +208,12 @@ def getTransactions(self) -> pd.DataFrame:
         Returns:
         db: pd.dataFrame - transactional database
         """
-        df = pd.DataFrame(self.db)
-        return df
+        column = "Transactions"
+        db = pd.DataFrame(columns=[column])
+        self.db = [tuple(x) for x in self.db]
+        for i in range(len(self.db)):
+            db.at[i,column] = self.db[i]
+        return db
 
 
 if __name__ == "__main__":