如果我运行以下代码:
from sklearn.datasets import load_breast_cancerimport lightgbm as lgbbreast_cancer = load_breast_cancer()data = breast_cancer.datatarget = breast_cancer.targetparams = { "task": "convert_model", "convert_model_language": "cpp", "convert_model": "test.cpp",}gbm = lgb.train(params, lgb.Dataset(data, target))
我期望会生成一个名为test.cpp
的文件,并将模型以C++格式保存。
然而,我的当前目录中没有任何文件生成。
我已经阅读了文档(https://lightgbm.readthedocs.io/en/latest/Parameters.html#io-parameters),但无法找出我做错的地方。
回答:
文档中提到:
注意:只能在CLI版本中使用
在convert_model和convert_model_language参数下方有这样的说明。
这意味着您可能应该使用LGBM的CLI(命令行界面)而不是Python封装来完成这个任务。
快速开始CLI版本的链接。