如何在 SMOTE（imblearn 模塊）之后提取新添加的行

Question

是否可以從 imblearn 的 smote function 創建的 pandas dataframe 中提取新添加的行？

Answer 1

我想我想通了。 顯然，它們被附加在 fit_resample 返回的 dataframe 的末尾：

我的目標是“死”

smotez = SMOTENC([10,11], random_state=555, k_neighbors=10)
smote_tomek = SMOTETomek(random_state=555, smote=smotez , n_jobs=-1)  
X_train_new, y_train_new = smote_tomek.fit_resample(X_train, y_train) 
train_data_new = pd.concat([X_train_new.iloc[1:],y_train_new],axis=1)
train_data_new.dropna(inplace=True)
smote_data = train_data_new.iloc[len(train_data)-1:,]
print("Y_train_smote:\n", npunique(smote_data['DIED']),smote_data['DIED'].mean())

如您所見，所有行都是少數 class（“DIED”）

Y_train_smote：[[1 91936]] 1.0

仔細檢查，下面的表達式應該返回 0：

print(len(smote_data) + len(X_train) - len(X_train_new))

0