search

BartekPog / modelcreator

Simple python package for creating predictive models
https://pypi.org/project/modelcreator/
MIT License
6 stars 0 forks source link

Add rare classes multiplication feature #7

Open BartekPog opened 4 years ago

BartekPog commented 4 years ago

We have to handle class imbalance. One of the methods is multiplying the rows with rare classes to equalize the class frequency in train dataset.

Add the proper method and expand the interface to handle this feature.

BartekPog commented 4 years ago

This can be achieved by using SMOTE or ADASYN algorithm. Both have already been implemented in Imblearn library.