This project is used to evaluate the performance of the oversampling method k-means SMOTE.
pip3 install -r requirements.txt
- Set local folder paths using
config.yml
(seeconfig.sample.yml
for an example). - Open
imbalanced_benchmark.py
to check and adaptexperiment_config
,classifiers
andoversampling_methods
. - Execute
imbalanced_benchmark.py
. - Once the script has run, open the results folder created with the current timestamp. It contains a pickled experiment (or CSV files for older versions of imbalanced-tools) which can be read by imbalanced-tools.
- Execute
plot_imbalanced_benchmark.py
with the timestamp as a parameter. This will generate a PDF file with plots in the experiment's folder.