Initial model implementation

include/model/model.hpp

@@ -0,0 +1,34 @@
+#ifndef CUDANET_MODEL_H
+#define CUDANET_MODEL_H
+
+#include <string>
+#include <vector>
+#include <map>
+#include "layer.cuh"
+
+namespace CUDANet {
+
+class Model {
+  public:
+    Model(const int inputSize, const int inputChannels);
+    ~Model();
+
+    float* predict(const float* input);
+
+    void addLayer(const std::string& name, Layers::SequentialLayer* layer);
+
+  private:
+
+    int inputSize;
+    int inputChannels;
+
+    int outputSize;
+    
+    std::vector<Layers::SequentialLayer*> layers;
+    std::map<std::string, Layers::WeightedLayer*> layerMap;
+
+};
+
+}  // namespace CUDANet
+
+#endif  // CUDANET_MODEL_H