Improvement predicting mental illness for health insurance

See merge request ki_lab/machine-learning-services!22