Imported from https://github.com/jsalt-coml/babytrain_multilabel