在Colab中使用PyCaret时无法调用XGBoost和CatBoost的解决方法

原因分析 #

PyCaret基础安装包不会自动包含XGBoost、CatBoost这类第三方模型的依赖，它们属于可选扩展组件，必须手动安装对应依赖后，才能在models()中显示并调用。

解决方案 #

1. 安装包含完整依赖的PyCaret版本 #

在Colab中执行以下命令，安装包含所有可选依赖的PyCaret预发布版本：

!pip install --pre pycaret[full]

注意：安装完成后必须重启运行时（通过Colab顶部菜单Runtime → Restart runtime操作），否则新安装的依赖不会生效。

2. 单独安装指定模型依赖 #

如果不需要完整依赖包，可仅安装需要的模型依赖：

• 安装XGBoost：

!pip install xgboost

• 安装CatBoost：

!pip install catboost

安装完成后同样需要重启运行时，再重新执行后续代码。

3. 验证修复效果 #

重启运行时后，重新执行你的代码：

from pycaret.regression import *
setup(data=data, target='price')
models()

此时models()的输出列表中应该能看到XGBoost、CatBoost等模型。

额外提示 #

• Colab重启运行时后，已安装的包会保留，但需要重新导入PyCaret库并执行setup()。
• 检查代码中是否误设了exclude_models参数，若有需移除对应模型的排除配置。

内容的提问来源于stack exchange，提问作者savasci