FIX: Label encoding done in compute_class_weight