Skip to main content

Geometric SMOTE algorithm.

Project description

Travis AppVeyor Codecov CircleCI ReadTheDocs PythonVersion Pypi Conda DOI Black

geometric-smote

Implementation of the Geometric SMOTE algorithm [1], a geometrically enhanced drop-in replacement for SMOTE. It is compatible with scikit-learn and imbalanced-learn.

Documentation

Installation documentation, API documentation, and examples can be found on the documentation.

Dependencies

geometric-smote is tested to work under Python 3.6+. The dependencies are the following:

  • numpy(>=1.1)
  • scikit-learn(>=0.21)
  • imbalanced-learn(>=0.4.3)

Additionally, to run the examples, you need matplotlib(>=2.0.0) and pandas(>=0.22).

Installation

geometric-smote is currently available on the PyPi’s repository and you can install it via pip:

pip install -U geometric-smote

The package is released also in Anaconda Cloud platform:

conda install -c algowit geometric-smote

If you prefer, you can clone it and run the setup.py file. Use the following commands to get a copy from GitHub and install all dependencies:

git clone https://github.com/AlgoWit/geometric-smote.git
cd geometric-smote
pip install .

Or install using pip and GitHub:

pip install -U git+https://github.com/AlgoWit/geometric-smote.git

Testing

After installation, you can use pytest to run the test suite:

make test

About

If you use geometric-smote in a scientific publication, we would appreciate citations to the following paper:

@article{Douzas2019,
  doi = {10.1016/j.ins.2019.06.007},
  url = {https://doi.org/10.1016/j.ins.2019.06.007},
  year = {2019},
  month = oct,
  publisher = {Elsevier {BV}},
  volume = {501},
  pages = {118--135},
  author = {Georgios Douzas and Fernando Bacao},
  title = {Geometric {SMOTE} a geometrically enhanced drop-in replacement for {SMOTE}},
  journal = {Information Sciences}
}

Classification of imbalanced datasets is a challenging task for standard algorithms. Although many methods exist to address this problem in different ways, generating artificial data for the minority class is a more general approach compared to algorithmic modifications. SMOTE algorithm [2], as well as any other oversampling method based on the SMOTE mechanism, generates synthetic samples along line segments that join minority class instances. Geometric SMOTE (G-SMOTE) is an enhancement of the SMOTE data generation mechanism. G-SMOTE generates synthetic samples in a geometric region of the input space, around each selected minority instance.

References:

[1]G. Douzas, F. Bacao, “Geometric SMOTE: a geometrically enhanced drop-in replacement for SMOTE”, Information Sciences, vol. 501, pp. 118-135, 2019.
[2]N. V. Chawla, K. W. Bowyer, L. O. Hall, W. P. Kegelmeyer, “SMOTE: synthetic minority over-sampling technique”, Journal of Artificial Intelligence Research, vol. 16, pp. 321-357, 2002.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for geometric-smote, version 0.1.3
Filename, size File type Python version Upload date Hashes
Filename, size geometric_smote-0.1.3-py3-none-any.whl (10.1 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size geometric-smote-0.1.3.tar.gz (9.3 MB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page