C++调用Python智能分类器

Python代码

import fastText
import jieba

# 判断小说类型的调用语句
def judge(novel):
    print(novel, " : ", classifier.predict(" ".join(jieba.cut(novel))))
    return str(classifier.predict(" ".join(jieba.cut(novel)))[0][0])

# classifier = fastText.train_supervised("data/novel_names.txt", lr=0.1, wordNgrams=1, loss="hs", epoch=20)
# model = classifier.save_model("data/novel_names.model")
classifier = fastText.load_model("data/novel_names.model")

C++代码（Qt为例）

#include <QCoreApplication>
#include <iostream>
#include <Python.h>
#include <QDebug>
using namespace std;

int main(int argc, char *argv[])
{
    QCoreApplication a(argc, argv);

    Py_Initialize();

    if (!Py_IsInitialized())
    {
        return -1;
    }

    PyObject* pModule = PyImport_ImportModule("wtnn"); // 模块名（不带后缀的文件名）
    if (!pModule)
    {
        printf("Can't open python file!\n");
        return -1;
    }

    PyObject* pFun = PyObject_GetAttrString(pModule, "judge"); // 函数名

    if (!pFun)
    {
        printf("Get function failed");
        return -1;
    }

    PyObject* args = Py_BuildValue("(s)", "测试"); // 构建参数，为元组形式
    PyObject* result = PyObject_CallObject(pFun, args);
    if (result)
    {
        char* str = nullptr;
        PyArg_Parse(result, "s", &str);
        qDebug() << str; // 可以直接printf，但是中文乱码，得用qDebug
    }

    Py_Finalize();

    return a.exec();
}

pro文件

INCLUDEPATH += -I C:\Users\HP\AppData\Local\Programs\Python\Python36\include
LIBS += -LC:\Users\HP\AppData\Local\Programs\Python\Python36\libs -lpython36