【发布时间】:2016-07-21 15:40:15
【问题描述】:
我正在尝试在我的 C++ 项目中使用 scikit learn。这是我正在使用的代码:
#include <Python.h>
PyObject* loadModule(char* name)//
{
PyObject* pName = PyString_FromString(name);
PyObject* pModule = PyImport_Import(pName);
Py_DECREF(pName);
return pModule;
}
void displayPyObject(PyObject* object)
{
PyObject* objectsRepresentation = PyObject_Repr(object);
PyErr_Print();
const char* s = PyString_AsString(objectsRepresentation);
PyErr_Print();
std::cout << "[ PYOBJECT ]" << s << std::endl;
}
//load functions/ attributes from module
PyObject* loadComponentFromModule(char* module, char* component)
{
PyObject* pModule = loadModule(module);
PyObject* pyComponent = PyObject_GetAttrString(pModule, component);
Py_DECREF(pModule);
return pyComponent;
}
//WRAPPER FOR KMEANS CLUSTERING FROM SCIKIT-LEARN
class KMeans
{
public:
KMeans(int nClusters)
{
PyObject* KmeansClass = loadComponentFromModule("sklearn.cluster", "KMeans");
PyObject* pName2 = PyInt_FromLong((long) nClusters);
PyObject* pArgs = PyTuple_New(1);
PyTuple_SetItem(pArgs, 0, pName2);
_Kcluster = PyObject_CallObject(KmeansClass, pArgs);
_closestor = loadComponentFromModule("sklearn.metrics","pairwise_distances_argmin_min");
Py_DECREF(KmeansClass);
Py_DECREF(pName2);
Py_DECREF(pArgs);
}
~KMeans()
{
Py_DECREF(_Kcluster);
Py_DECREF(_closestor);
}
void setNumClusters(int nClusters)
{
std::cout << "change to number cluster: " << nClusters << "\n";
PyObject* nCluster = PyInt_FromLong((long) nClusters);
int code = PyObject_SetAttrString(_Kcluster,"n_clusters", nCluster);
PyErr_Print();
if (code == -1)
{
std::cout << "[Error] KMeans.setNumClusters() Failed!! - Number of clusters didn't change!!\n";
}
Py_DECREF(nCluster);
}
void info()
{
displayPyObject(_Kcluster);
}
private:
PyObject* _Kcluster;
//PyObject* _result;
PyObject* _closestor;
};
PyObject* loadClassifier()
{
PyObject* loader = loadComponentFromModule("sklearn.externals.joblib", "load");
PyObject* pName2 = PyString_FromString("lda.pkl");
PyObject* pArgs = PyTuple_New(1);
PyTuple_SetItem(pArgs, 0, pName2);
PyObject* clf = PyObject_CallObject(loader, pArgs);
Py_DECREF(loader);
Py_DECREF(pName2);
Py_DECREF(pArgs);
// displayPyObject(clf);
return clf;
}
void produce_error()
{
std::cout << "============================= LINE 0 =========================================\n";
PyObject* clf = loadClassifier();//"sklearn.externals.joblib", "load");
std::cout << "============================= LINE 1 =========================================\n";
KMeans cluster(8);
std::cout << "============================= LINE 2 =========================================\n";
cluster.setNumClusters(5);
std::cout << "============================= LINE 3 =========================================\n";
cluster.info();
std::cout << "============================= LINE 4 =========================================\n";
}
int main(int argc, char *argv[])
{
Py_Initialize();
produce_error();
Py_Finalize();
return 0;
}
每次我运行这个程序,我都会得到错误:
============================= LINE 0 =========================================
============================= LINE 1 =========================================
Segmentation fault (core dumped)
据我了解,在不调用分类器的情况下,我没有收到任何错误,但是每当我调用它 (loadClassifier()) 并创建 KMeans 实例时,都会显示错误。有时错误在我加载模块的 KMeans 的构造函数内部,在另一个函数中使用相同的代码(不在此处),错误在 PyObject_Repr() 内部(在 displayPyObject() 内部)。
以前有人遇到过同样的问题吗?你知道怎么解决吗?提前致谢。
【问题讨论】:
标签: c++ segmentation-fault python-c-api