From 098fd65074ef4157f0a0b9f47893ea87439b27c5 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Tue, 28 May 2024 18:42:58 +0200
Subject: [PATCH] Add model summary

---
 include/model/model.hpp | 2 ++
 src/model/model.cpp     | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/include/model/model.hpp b/include/model/model.hpp
index 815298d..9933e35 100644
--- a/include/model/model.hpp
+++ b/include/model/model.hpp
@@ -39,6 +39,8 @@ class Model {
 
     bool validate();
 
+    void printSummary();
+
   protected:
     Layers::Input*  inputLayer;
     Layers::Output* outputLayer;
diff --git a/src/model/model.cpp b/src/model/model.cpp
index 9319e1d..f493d22 100644
--- a/src/model/model.cpp
+++ b/src/model/model.cpp
@@ -186,4 +186,12 @@ bool Model::validate() {
     }
 
     return valid;
+}
+
+void Model::printSummary() {
+    std::cout << "Model summary:" << std::endl;
+    for (const auto& layer : layers) {
+        std::cout << "Layer: " << layer.first << ", input size: " << layer.second->getInputSize() << ", output size: "
+                  << layer.second->getOutputSize() << std::endl;
+    }
 }
\ No newline at end of file