commit a7bf9066c7cac609135db72e3e64bfaefe5ac26e
parent 21502d283cf2bc3a11de7ba25f5df15bd6c1e634
Author: David Freifeld <freifeld.david@gmail.com>
Date: Tue, 30 Jun 2020 17:15:51 -0700
Loose boilerplate for CNN
Diffstat:
| A | cnn.cpp | | | 30 | ++++++++++++++++++++++++++++++ |
1 file changed, 30 insertions(+), 0 deletions(-)
diff --git a/cnn.cpp b/cnn.cpp
@@ -0,0 +1,30 @@
+#include "bpnn.hpp"
+#include "utils.hpp"
+
+class ConvLayer
+{
+ Eigen::MatrixXd* input;
+ Eigen::MatrixXd* kernel;
+ Eigen::MatrixXd* output;
+};
+class PoolingLayer
+{
+};
+
+class ConvNet : Network
+{
+public:
+ ConvNet(char* path, int batch_sz, float learn_rate, float bias_rate);
+};
+
+ConvNet::ConvNet(char* path, int batch_sz, float learn_rate, float bias_rate)
+{
+ learning_rate = learn_rate;
+ bias_lr = bias_rate;
+ instances = prep_file(path, "./shuffled.txt");
+ length = 0;
+ t = 0;
+ batch_size = batch_sz;
+ data = fopen("./shuffled.txt", "r");
+ batches = 0;
+}