Add torch library

2023-10-31 10:07:24 +01:00
parent 77c33942f6
commit cb3281ed91
16 changed files with 884 additions and 27 deletions
--- a/src/PyClassifier.h
+++ b/src/PyClassifier.h
@@ -12,8 +12,7 @@ namespace pywrap {
        PyClassifier(const std::string& module, const std::string& className);
        virtual ~PyClassifier();
        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);
+        std::string callMethod(const std::string& method);
    private:
        PyWrap* pyWrap;
        std::string module;