Allowing for custom loss functions to be used (e.g. focal loss)

This MR adds focal loss to the repo as well as the possibility to add more custom loss functions.

For information on focal loss and how to use it see arXiv:1708.02002.