Starting on the model.h, but need to make layer structs and structs for loss and optimizers

2026-05-30 09:13:52 +02:00
parent cb65174cf4
commit edad247227
30 changed files with 1879 additions and 159 deletions
@@ -0,0 +1,30 @@
+#pragma once
+
+#include "core/omp_config.h"
+
+#include "utils/vector.h"
+#include "utils/matrix.h"
+
+#include "modules/neural_networks/layers/Layer.h"
+
+
+namespace neural_networks {
+
+    template <typename T>
+    struct Model {
+
+        std::vector<Layer<T>*> layers;
+
+        Model() = default;
+
+        void add(Layer<T>& layer) {
+            layers.push_back(&layer);
+        }
+
+    };
+
+ 
+
+
+
+} // end namespace neural_networks