了解过机器学习的人应该都知道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语言代码了。
标签:机器,AI,模型,C++,python,joblib,clf From: https://www.cnblogs.com/ExMan/p/17211473.html