commit 53bab8ea1ab08422d3137c9181726367c7d3b451
parent 384b86c4270d0f4d32242d90a530919f5c17218e
Author: David Freifeld <freifeld.david@gmail.com>
Date: Fri, 25 Sep 2020 16:59:34 -0700
Small tweaks
Diffstat:
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/capsnet.cpp b/src/capsnet.cpp
@@ -1,6 +1,7 @@
#include "utils.hpp"
#include "cnn.hpp"
+#include "bpnn.hpp"
class Capsule
{
@@ -18,6 +19,5 @@ public:
void add_conv_layer();
void add_capsule();
void feedforward();
- void backpropagate();
void routing();
};
diff --git a/src/cnn.hpp b/src/cnn.hpp
@@ -1,3 +1,6 @@
+#ifndef CNN_H
+#define CNN_H
+
class ConvLayer
{
public:
@@ -47,3 +50,4 @@ public:
void set_label(Eigen::MatrixXf newlabels);
void initialize();
};
+#endif /* MODULE_H */