Imbalanced_classification_task

The code of this warehouse is mainly dedicated to solving the problem of uneven sample distribution in binary classification tasks, where there is a significant difference in the number of samples between positive and negative categories.

Install Packages

1. pip install torch timm sklearn matplotlib searborn

TRAIN AND EVALUATE MODEL

1. cd Imbalanced_classification_task
2. python train_gpu.py