Focal Loss of multi-classification in tensorflow - View it on GitHub