Begin predict
This commit is contained in:
@@ -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;
|
||||
|
Reference in New Issue
Block a user