Weighted categorical cross entropy

Question

please I\'m trying to build an NLP classifier on top of BERT but I\'m struggling with data imbalance. I\'m looking for an implementation of weighted CategoricalCrossEntropy.