* Added focal loss support to single label tasks. * Updated focal loss docstring
ClassBalancedDataset