使用lgb.cv时出现ValueError: Supported target types are: (‘binary’, ‘multiclass’). Got ‘continuous’ instead.
默认情况下,lightgbm.cv中的stratify参数是True。 根据the documentation:
stratified (bool, optional (default=True)) – Whether to perform stratified sampling.
但是stratify只处理分类问题。因此,要处理回归,需要使其为假。
cv_results = lgb.cv(
params,
dftrainLGB,
num_boost_round=100,
nfold=3,
metrics=‘mae’,
early_stopping_rounds=10,
# This is what I added
**stratified=False**
)
现在它开始工作了。