Add id to classifier
This commit is contained in:
@@ -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 */
|
||||
|
Reference in New Issue
Block a user