### Description of feature If a sample comes from some sort of disease, there should be a way to inform the model about that.