【发布时间】:2020-07-14 12:42:03
【问题描述】:
求助,这个简单的代码会产生一个奇怪的错误。
from sklearn.feature_selection import RFECV
from xgboost.sklearn import XGBRegressor
import sklearn.metrics
from sklearn.metrics import mean_absolute_error
estimator = XGBRegressor()
selector = RFECV(estimator, step=1, min_features_to_select=1, cv=10, scoring='neg_mean_absolute_error')
selector = selector.fit(x, y.values.flatten())
我的回归器已经运行顺利,但 selector.fit 没有。 对于 RFE 或 RFECV,我得到相同的结果:
~/miniconda2/envs/py3/lib/python3.6/site-packages/xgboost/sklearn.py in coef_(self)
714 .format(self.booster))
715 b = self.get_booster()
--> 716 coef = np.array(json.loads(b.get_dump(dump_format='json')[0])['weight'])
717 # Logic for multiclass classification
718 n_classes = getattr(self, 'n_classes_', None)
KeyError: 'weight'
提前致谢。
【问题讨论】:
-
您能否将您的代码重新格式化为 20 个字符的行宽?我无法在超暗模式下阅读它。
标签: python scikit-learn xgboost feature-selection