Add id to classifier

This commit is contained in:
2023-11-10 11:11:10 +01:00
parent 384b9071a2
commit 55286168cb
6 changed files with 56 additions and 46 deletions

View File

@@ -10,6 +10,7 @@
#include <torch/torch.h>
#include "PyWrap.h"
#include "Classifier.h"
#include "TypeId.h"
namespace pywrap {
class PyClassifier : public Classifier {
@@ -29,7 +30,7 @@ namespace pywrap {
PyWrap* pyWrap;
std::string module;
std::string className;
uint32_t id;
clfId_t id;
bool fitted;
};
} /* namespace pywrap */