LightGBMError:不支持功能名称中的特殊JSON字符相同的代码在jupyter中工作,但在Spyder中不工作

2024-09-27 07:25:48 发布

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

我有以下代码:

    most_important = features_importance_chi(importance_score_tresh, 
    df_user.drop(columns = 'CHURN'),churn)
    X = df_user.drop(columns = 'CHURN')
    churn[churn==2] = 1
    y = churn

    # handle undersample problem
    X,y = handle_undersampe(X,y)

    # train the model

    X=X.loc[:,X.columns.isin(most_important)].values
    y=y.values

    parameters = {
    'application': 'binary',
    'objective': 'binary',
    'metric': 'auc',
    'is_unbalance': 'true',
    'boosting': 'gbdt',
    'num_leaves': 31,
    'feature_fraction': 0.5,
    'bagging_fraction': 0.5,
    'bagging_freq': 20,
    'learning_rate': 0.05,
    'verbose': 0
    }

    # split data
    x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42, stratify=y)

    train_data = lightgbm.Dataset(x_train, label=y_train)
    test_data = lightgbm.Dataset(x_test, label=y_test)
    model = lightgbm.train(parameters,
                       train_data,
                       valid_sets=[train_data, test_data], 
                       **feature_name=most_important,**
                       num_boost_round=5000,
                       early_stopping_rounds=100) 

和返回最重要参数的函数

def features_importance_chi(importance_score_tresh, X, Y):
    model = ExtraTreesClassifier(n_estimators=10)
    model.fit(X,Y.values.ravel())
    feature_list = pd.Series(model.feature_importances_,
                             index=X.columns)
    feature_list = feature_list[feature_list > importance_score_tresh]
    feature_list = feature_list.index.values.tolist()
    return feature_list

有趣的是Spyder中的代码返回以下错误

LightGBMError: Do not support special JSON characters in feature name.

但在朱皮特很好用。我能够打印最重要功能的列表

你知道这个错误的原因是什么吗


Tags: columnstestmostdatamodeltrainfeaturelist
1条回答
网友
1楼 · 发布于 2024-09-27 07:25:48

你知道吗,这条消息经常出现在LGBMClassifier()模型上,即LGBM。 一旦你上传了熊猫的数据,并且你的头有问题,只要把这一行放在开头就行了:

import re
df = df.rename(columns = lambda x:re.sub('[^A-Za-z0-9_]+', '', x))

相关问题 更多 >

    热门问题