Анаконда проблемы с py-xgboost и sklearn

Question

У меня установлена ​​Anaconda 5.3.0 Winx86.Я установил пакет xgboost, используя
conda install -c anaconda py-xgboost. В Jupyter Notebook я могу импортировать XGBRegressor
from xgboost import XGBRegressor Но когда я использую xgb = XGBRegressor().Я получаю следующую ошибку:

XGBoostError                              Traceback (most recent call last)
<ipython-input-2-99b3db9fba2c> in <module>
----> 1 xgb = XGBRegressor()

C:\ProgramData\Anaconda3\lib\site-packages\xgboost\sklearn.py in __init__(self, max_depth, learning_rate, n_estimators, silent,objective,nthread, gamma, min_child_weight, max_delta_step, subsample, colsample_bytree, colsample_bylevel, reg_alpha, reg_lambda, scale_pos_weight, base_score, seed, missing)
119                  base_score=0.5, seed=0, missing=None):
120         if not SKLEARN_INSTALLED:
--> 121     raise XGBoostError('sklearn needs to be installed in order to use this module')
122         self.max_depth = max_depth
123         self.learning_rate = learning_rate

XGBoostError: sklearn needs to be installed in order to use this module

Но я могу импортировать модуль sklearn без ошибок: import sklearn.Я попытался переустановить anaconda и xgboost, но ничего не получилось.

Answer 1

Установите sklearn с помощью pip.

sudo pip install sklearn

Теперь импортируйте XGBRegressor в код:

from xgboost.sklearn import XGBRegressor