Begin templating

src/PyClassifier.h

@@ -1,6 +1,9 @@
 #ifndef PYCLASSIFER_H
 #define PYCLASSIFER_H
 #include <string>
+#include <map>
+#include <vector>
+#include <torch/torch.h>
 #include "PyWrap.h"
 
 namespace pywrap {
@@ -8,7 +11,9 @@ namespace pywrap {
     public:
         PyClassifier(const std::string& module, const std::string& className);
         virtual ~PyClassifier();
-        void callMethod(const std::string& method);
+        PyClassifier& fit(torch::Tensor& X, torch::Tensor& y, const std::vector<std::string>& features, const std::string& className, std::map<std::string, std::vector<int>>& states);
+        template <typename T>
+        T callMethod(const std::string& method);
     private:
         PyWrap* pyWrap;
         std::string module;