I am working on epilepsy seizure prediction I have 2 imbalanced classes , class 0 and class 1. I wan to use focal loss to balance my dataset. I used keras in my work. I searche