目标编码 def gen_target_encoding_feats(train, train_2, test, encode_cols, target_col, n_fold=10): '''生成target encoding特征''' # for training set - cv tg_feats = np.zeros((train.shape[0], len(encode_cols))) kfold = StratifiedKFold(n_splits=n_fold, random_state=1024, shuffle=True) for _, (train_index, val_index) in enumerate(kfold.split(train[encode_cols], train[target_col])): df_train, df_val = train.iloc[train_index], train.iloc[val_index] for idx, col in enumerate(encode_cols): target_mean_dict = df_train.groupby(col)[target_col].mean() if not df_val[f'{col}_mean_target'].empty: df_val[f'{col}_mean_target'] = df_val[col].map(target_mean_dict) tg_feats[val_index, idx] = df_val[f'{col}_mean_target'].values for idx, encode_col in enumerate(encode_cols): train[f'{encode_col}_mean_target'] = tg_feats[:, idx] # for train_2 set - cv tg_feats = np.zeros((train_2.shape[0], len(encode_cols))) kfold = StratifiedKFold(n_splits=n_fold, random_state=1024, shuffle=True) for _, (train_index, val_index) in enumerate(kfold.split(train_2[encode_cols], train_2[target_col])): df_train, df_val = train_2.iloc[train_index], train_2.iloc[val_index] for idx, col in enumerate(encode_cols): target_mean_dict = df_train.groupby(col)[target_col].mean() if not df_val[f'{col}_mean_target'].empty: df_val[f'{col}_mean_target'] = df_val[col].map(target_mean_dict) tg_feats[val_index, idx] = df_val[f'{col}_mean_target'].values for idx, encode_col in enumerate(encode_cols): train_2[f'{encode_col}_mean_target'] = tg_feats[:, idx] # for testing set for col in encode_cols: target_mean_dict = train.groupby(col)[target_col].mean() test[f'{col}_mean_target'] = test[col].map(target_mean_dict) return train, train_2, test features = ['house_exist', 'debt_loan_ratio', 'industry', 'title'] train_1, train_2, test = gen_target_encoding_feats(train_1, train_2, test, features, ['isDefault'], n_fold=10) 出现报错ValueError: The truth value of a DataFrame is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().在哪里修改怎么修改
时间: 2024-01-27 20:03:39 浏览: 57
random_runner:我为KDD Cup 2015设计的解决方案
在该代码中,报错出现在第11行,即 if not df_val[f'{col}_mean_target'].empty: 这一行。这里的问题是在判断一个 DataFrame 是否为空时出现了歧义。正确的判断方式应该是使用 df_val[f'{col}_mean_target'].empty 来判断该 DataFrame 是否为空,而不是使用 not。因此,你可以将该行修改为 if df_val[f'{col}_mean_target'].empty:,即将 not 去掉即可。
阅读全文