当lightgbm的标签为(0,1)之间的连续值时

2024-09-28 01:31:09 发布

您现在位置:Python中文网/ 问答频道 /正文

ValueError: Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-34-6adcad511f10> in <module>()
     25 kf = skf.split(train_data[feature_name],train_data['RULR__mean'])
     26 # 交叉验证
---> 27 for index, (train_index, test_index) in enumerate(kf):
     28     print(index)
     29     X_train, X_valid, y_train, y_valid = train_data[train_index][feature_name], train_data[test_index][feature_name],train_data[train_index][ 'RULR__mean'], train_data[test_index][ 'RULR__mean']



3 frames

/usr/local/lib/python3.6/dist-packages/sklearn/model_selection/_split.py in _make_test_folds(self, X, y)
    639             raise ValueError(
    640                 'Supported target types are: {}. Got {!r} instead.'.format(
--> 641                     allowed_target_types, type_of_target_y))
    642 
    643         y = column_or_1d(y)

ValueError: Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

Tags: nameintesttargetdataindextrainare

热门问题