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

refactor folders

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-11-11 10:52:35 +01:00
parent 74fb0968c7
commit b6a3a05020
9 changed files with 119 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/PyWrap.cc
View File

@@ -142,24 +142,22 @@ namespace pywrap {
{
PyObject* instance = getClass(id);
CPyObject result;
std::string method = "fit";
CPyObject method = PyUnicode_FromString("fit");
try {
if (!(result = PyObject_CallMethodObjArgs(instance, PyUnicode_FromString(method.c_str()), X.getObject(), y.getObject(), NULL)))
if (!(result = PyObject_CallMethodObjArgs(instance, method.getObject(), X.getObject(), y.getObject(), NULL)))
errorAbort("Couldn't call method fit");
}
catch (const std::exception& e) {
errorAbort(e.what());
}
// Py_XDECREF(result);
}
PyObject* PyWrap::predict(const clfId_t id, CPyObject& X)
{
PyObject* instance = getClass(id);
PyObject* result;
std::string method = "predict";
CPyObject method = PyUnicode_FromString("predict");
try {
if (!(result = PyObject_CallMethodObjArgs(instance, PyUnicode_FromString(method.c_str()), X.getObject(), NULL)))
if (!(result = PyObject_CallMethodObjArgs(instance, method.getObject(), X.getObject(), NULL)))
errorAbort("Couldn't call method predict");
}
catch (const std::exception& e) {
@@ -172,16 +170,15 @@ namespace pywrap {
{
PyObject* instance = getClass(id);
CPyObject result;
std::string method = "score";
CPyObject method = PyUnicode_FromString("score");
try {
if (!(result = PyObject_CallMethodObjArgs(instance, PyUnicode_FromString(method.c_str()), X.getObject(), y.getObject(), NULL)))
if (!(result = PyObject_CallMethodObjArgs(instance, method.getObject(), X.getObject(), y.getObject(), NULL)))
errorAbort("Couldn't call method score");
}
catch (const std::exception& e) {
errorAbort(e.what());
}
double resultValue = PyFloat_AsDouble(result);
// Py_XDECREF(result);
return resultValue;
}
}