Multi-class classification with focal loss for imbalanced datasets - View it on GitHub