jacobian

commit 267aed8b3c15b379727dadaa157cc035fbf4a436
parent 713a8996738b928e3049057a791b1531c5f93bc8
Author: David Freifeld <freifeld.david@gmail.com>
Date:   Fri, 25 Sep 2020 10:51:31 -0700

Added beginning of capsule networks

Diffstat:
A
src/capsnet.cpp
 | 
23
+++++++++++++++++++++++

1 file changed, 23 insertions(+), 0 deletions(-)
diff --git a/src/capsnet.cpp b/src/capsnet.cpp
@@ -0,0 +1,23 @@
+
+#include "utils.hpp"
+#include "cnn.hpp"
+
+class Capsule
+{
+    Capsule* children;
+    Capsule* parents;
+}
+
+class CapsNet
+{
+public:
+    std::vector<ConvLayer> conv_layers;
+    std::vector<Capsule> primary_caps;
+    std::vector<Capsule> class_caps;
+    CapsNet(char* path);
+    void add_conv_layer();
+    void add_capsule();
+    void feedforward();
+    void backpropagate();
+    void routing();
+};