Querying of a single MNIST sample

gui/NeuroUI/mnistloader.cpp

@@ -8,6 +8,21 @@ void MnistLoader::load(const std::string &databaseFileName, const std::string &l
     loadLabels(labelsFileName);
 }
 
+size_t MnistLoader::getSamleCount() const
+{
+    return samples.size();
+}
+
+const MnistLoader::MnistSample &MnistLoader::getSample(size_t index) const
+{
+    if (index >= samples.size())
+    {
+        throw std::runtime_error("MNIST sample index out of range");
+    }
+
+    return *(samples[index].get());
+}
+
 const MnistLoader::MnistSample &MnistLoader::getRandomSample() const
 {
     size_t sampleIndex = (std::rand() * (samples.size() - 1)) / RAND_MAX;

gui/NeuroUI/mnistloader.h

@@ -31,6 +31,8 @@ private:
 public:
     void load(const std::string &databaseFileName, const std::string &labelsFileName);
 
+    size_t getSamleCount() const;
+    const MnistSample &getSample(size_t index) const;
     const MnistSample &getRandomSample() const;
 
 private: