Maybe say that this only applies to multi-label or multi-output prediction (does the current implementation work for multi-output?)