本帖最后由 ultraman 于 2020-1-2 10:27 编辑

本文尝试将PC的xgboost预测模型移植到MCU开发板上
Extreme Gradient Boosting(XGBoost)是由华盛顿大学(University of Washington)的陈天奇作为Distributed (Deep) Machine Learning Community (DMLC) 组员所开发的一个研究项目。在陈天奇与队友一起赢得了Higgs Machine Learning Challenge后,许多的数据科学竞赛队伍使用XGBoost并获得了冠军,促进了该工具在数据科学应用中的广泛使用。
XGBOOST是一个监督模型,xgboost对应的模型本质是一堆CART树。用一堆树做预测,就是将每棵树的预测值加到一起作为最终的预测值。
因为XGBOOST本身太大,因此将模型文件通过代码转换转为C++代码,利用编译器优化执行.
1.png
XGBOOST的文本输出模型格式
2.png
XGBOOT文本模型的提取
3.png
XGBOOST树模型转换为C代码
4.png
转换后的树模型
5.png
利用编译器编译为目标代码,下载到板上运行
补充xgboost模型转换代码 model2cpp.py.zip (1.19 KB, 下载次数: 16)
全部回复 3
  • 17 主题
  • 561 帖子
  • 3028 积分
身份:LV5 资深技术员
E币:2244
强人呀!
  • 0 主题
  • 1711 帖子
  • 6179 积分
身份:LV6 初级工程师
E币:2180
下载到板上运行
  • 0 主题
  • 1 帖子
  • 225 积分
身份:LV1 技术小白
E币:219
您好,请问这个转换文件应该怎么使用?
回复楼主
您需要登录后才可以评论 登录 立即注册