From c5d5e02f00348014ca1a978952d07eee26b535f2 Mon Sep 17 00:00:00 2001
From: thibault allenet <thibault.allenet@cea.fr>
Date: Wed, 6 Dec 2023 11:08:18 +0000
Subject: [PATCH] Generic abstract database class
---
include/aidge/data/Database.hpp | 34 ++++++++++++++++-----------------
1 file changed, 16 insertions(+), 18 deletions(-)
diff --git a/include/aidge/data/Database.hpp b/include/aidge/data/Database.hpp
index 96292f242..04aa008a1 100644
--- a/include/aidge/data/Database.hpp
+++ b/include/aidge/data/Database.hpp
@@ -2,40 +2,38 @@
#define Database_H_
#include <cstring>
+#include <memory>
+#include <vector>
+#include <tuple>
#include "aidge/data/Tensor.hpp"
-namespace Aidge{
+namespace Aidge {
/**
- * @brief Database. An abstract class representing a database. All databases should inherit from this class. All subclasses should overwrite :cpp:function:`Database::get_item` to fetch data from a given index.
- * @todo Make the dataset generic. Always ground truth.
+ * @brief Database. An abstract class representing a map from a key to data. All databases should inherit from this class. All subclasses should overwrite :cpp:function:`Database::getItem` to fetch data from a given index.
*/
class Database {
public:
-
+ Database() = default;
virtual ~Database() = default;
/**
- * @brief Fetch a data sample and its corresponding ground_truth
- * @param index index of the pair (```data```, ```ground truth```) to fetch from the database
- * @return A pair of pointers to the data (first) and its corresping ground truth (second)
+ * @brief Fetch an item of the database.
+ * @param index index of the item.
+ * @return vector of data mapped to index.
*/
- virtual std::pair<std::shared_ptr<Tensor>,std::shared_ptr<Tensor>> get_item(unsigned int index) = 0;
+ virtual std::vector<std::shared_ptr<Tensor>> getItem(std::size_t index) = 0;
- /**
- * @return The number of data samples in the database.
+ /**
+ * @brief Get the number of items in the database
+ *
+ * @return std::size_t
*/
- virtual unsigned int get_len() = 0;
-
-protected:
-
- std::vector<std::shared_ptr<Tensor>> mData;
- std::vector<std::shared_ptr<Tensor>> mLabel;
+ virtual std::size_t getLen() = 0;
};
-
-}
+} // namespace Aidge
#endif /* Database_H_ */
\ No newline at end of file
--
GitLab