Using machine learning for medical solutions