diff --git a/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py b/PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py index 127663c6..25d7a3a3 100644 --- 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__":