Python上训练好机器学习模型后转成C或C++
·
在Python上训练好机器学习模型后可使用m2cgen(Model 2 Code Generator)将训练好的机器学习模型转换为独立的、可移植的代码。m2cgen(Model 2 Code Generator)是一个开源工具,它支持多种流行的机器学习库和框架,包括 scikit-learn、XGBoost、LightGBM、TensorFlow 等,并能够将这些模型转换为多种编程语言的代码,如 C、C++、Go、Java、JavaScript、Python、Rust、Visual Basic .NET 等。
首先安装好m2cgen
pip install m2cgen
然后将训练好的模型转成目标个数,以C为例:
import m2cgen as m2c
# 转成 C 代码
code = m2c.export_to_c(model)
# 将C 代码保存到文件
with open("model_name.c", "w") as f:
f.write(code)
更多推荐

所有评论(0)