rmontanana/PyWrap
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding tensor methods

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-11-01 14:13:45 +01:00
parent cb3281ed91
commit 296ed6b785
9 changed files with 144 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/PyClassifier.h
View File

@@ -12,7 +12,10 @@ 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);
std::string callMethod(const std::string& method);
torch::Tensor predict(torch::Tensor& X);
double score(torch::Tensor& X, torch::Tensor& y);
std::string version();
std::string callMethodString(const std::string& method);
private:
PyWrap* pyWrap;
std::string module;