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

Working fit/score

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-11-06 00:46:26 +01:00
parent f4928386bb
commit 3e92372d1c
7 changed files with 65 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/PyClassifier.cc
View File

@@ -54,10 +54,10 @@ namespace pywrap {
{
std::cout << "PyClassifier:fit:Converting X to PyObject" << std::endl;
auto [Xn, yn] = tensors2numpy(X, y);
CPyObject Xp = Xn.ptr();
CPyObject Xp = boost::python::incref(boost::python::object(Xn).ptr());
std::cout << "PyClassifier:fit:Converting y to PyObject" << std::endl;
print_array(yn);
CPyObject yp = yn.ptr();
CPyObject yp = boost::python::incref(boost::python::object(yn).ptr());
std::cout << "PyClassifier:fit:Calling fit" << std::endl;
pyWrap->fit(module, this->className, Xp, yp);
return *this;
@@ -65,18 +65,18 @@ namespace pywrap {
torch::Tensor PyClassifier::predict(torch::Tensor& X)
{
auto Xn = tensor2numpy(X);
print_array(Xn);
CPyObject Xp = Xn.ptr();
CPyObject Xp = boost::python::incref(boost::python::object(Xn).ptr());
auto PyResult = pyWrap->predict(module, className, Xp);
auto result = torch::tensor({ 1,2,3 });
return result;
}
double PyClassifier::score(torch::Tensor& X, torch::Tensor& y)
{
std::cout << "PyClassifier::Score:Converting X to PyObject" << std::endl;
auto [Xn, yn] = tensors2numpy(X, y);
CPyObject Xp = Xn.ptr();
CPyObject yp = yn.ptr();
CPyObject Xp = boost::python::incref(boost::python::object(Xn).ptr());
CPyObject yp = boost::python::incref(boost::python::object(yn).ptr());
print_array(yn);
auto result = pyWrap->score(module, className, Xp, yp);
return result;