XGBoost 找不到 sklearn

Question

我正在尝试 XGBoost，但被我无法弄清楚的错误阻止。我在活动环境中安装了 sklearn，可以通过在同一个 notebook 中训练 sklearn RandomForestClassifier 来验证它。当我尝试训练 XGBoost 模型时出现错误XGBoostError: sklearn needs to be installed in order to use this module

这有效：

clf = RandomForestClassifier(n_estimators=200, random_state=0, n_jobs=-1)

这会引发异常：

clf = xgb.XGBClassifier(max_depth=3, n_estimators=300, learning_rate=0.05).fit(train_X, train_y)

更新：使用完全相同的代码和导入创建了一个 PyCharm 模块，并且它无一例外地执行。所以这似乎是一个 Jupyter Notebook 问题。PyCharm 指向与笔记本相同的 Anaconda 环境。

更新 2：创建一个新笔记本并从引发异常的笔记本中复制代码。代码在新笔记本中运行正常。叹。案件结案。

Answer 1

遇到同样的问题，我在jupyter notebook 运行时安装sklearn 后安装。通过重新启动我的 Jupyter 笔记本服务器，能够找到安装。xgboost xgboostsklearn

在我安装sklearn 之前安装的另一个新环境中对此进行了测试，xgboost 然后启动我的 jupyter notebook 没有问题。