Begin predict

This commit is contained in:
2023-11-06 15:22:27 +01:00
parent 3e92372d1c
commit ed21f90b69
8 changed files with 55 additions and 69 deletions

View File

@@ -10,7 +10,6 @@
#include "PyWrap.h"
namespace pywrap {
class PyClassifier {
public:
PyClassifier(const std::string& module, const std::string& className);
@@ -19,7 +18,6 @@ namespace pywrap {
torch::Tensor predict(torch::Tensor& X);
double score(torch::Tensor& X, torch::Tensor& y);
std::string version();
std::string graph();
std::string callMethodString(const std::string& method);
private:
PyWrap* pyWrap;