similar as doen for training the disease dataset.
similar as doen for training the disease dataset.