divide the base model based on classes not instances