首页 > 解决方案 > Catboost:更小的安装可能吗?

问题描述

有没有办法以更小的占地面积来部分安装 CatBoost?我拥有的 catboost 版本占用了 161MB,我正在尝试查看是否可以减小大小。

对于 XGB,我可以通过运行“python -m pip install --no-binary xgboost xgboost”来安装没有二进制文件(并且没有 gpu 支持),这将 XGB 的大小从 400MB 减少到 10MB。CatBoost 有类似的东西吗?

提前致谢!

标签: catboost

解决方案


根据他们的文档,他们有一个仅 C++ 标头仅预测版本,我猜这是它可能用于服务目的的最小版本: https://github.com/catboost/catboost/blob/master/catboost/CatboostModelAPI。 MD

如果您只关心服务,那么也可以将模型导出为 ONNX 等格式并使用不同的库读取它,而无需安装 catboost。

如果您想要一个也可用于培训的小型 catboost 库,那么您可能不走运。


推荐阅读