AI机器学习模型python到C/C++的转换播

了解过机器学习的人应该都知道python的sklearn库非常好用的机器学习助手。从sklearn导入某个机器学习的库,调用fit函数即可生成模型,用来预测测试数据。

1、保存模型

如果想将模型保存下来,可以使用dump函数保存,往往会保存为.m或者.pkl文件,如下:

joblib.dump(clf, "ArrhythmiaClassifier.m", compress=0)

2、加载模型

下次想用使用这个模型时,只需要用joblib.load函数导入即可,如下:

clf = joblib.load(".//trainModel_twoClass.m")

3、模型的语言转换

实际工程开发中,往往用到的不是python语言,而是C/C++,或者java,甚至是JS,那怎么把模型从python转换为目标语言呢?其实也很简单,只需要下面四句话,以C语言为例:

porter = Porter(clf, language='c')

output = porter.export(embed_data=True)

with open('ArrhythmiaClassifier.cpp', 'w') as f:

f.write(output)

如此,就将模型转换为对应的c语言代码了。

posted on 2023-03-13 15:01  ExplorerMan  阅读(656)  评论(0)    收藏  举报

导航