Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/random_undersampling_imblearn.py

Created October 13, 2021 17:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save liannewriting/d4623a1ed40c4f4be9a10de1ebec755d to your computer and use it in GitHub Desktop.
Save liannewriting/d4623a1ed40c4f4be9a10de1ebec755d to your computer and use it in GitHub Desktop.
Download ZIP
python imbalanced data machine learning classification
Raw
random_undersampling_imblearn.py
from imblearn.under_sampling import RandomUnderSampler
rus = RandomUnderSampler(random_state=888)
X_resampled, y_resampled = rus.fit_resample(df_train[features], df_train['Class'])
y_resampled.value_counts()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment