Identifying diseases that cause psychological trauma and social avoidance by Xgboost

With the development of modern medical technology, the survival time of patients with most diseases is significantly prolonged. Changes in the physical, psychological, and social functions that occur after treatment and during treatment are a serious problem for patients and their families. Under the guidance of bio-psycho-social medical model theory, patients need not only disease care, but also mental health. Therefore, identifying diseases that can cause psychological trauma and social avoidance will provide great guidance for medical staff and greatly improve the quality of life of patients. In this paper, Xgboost is introduced to identify diseases that cause psychological trauma and social avoidance. Firstly, the diseases are known to cause psychological problems are extracted from literatures, and then the relevant diseases are extracted in the Disease Ontology (DO), and finally the proteins corresponding to each disease are obtained from Uniprot. We encoded proteins as the feature of each disease and put them into Xgboost. Finally, we identified the diseases that cause psychological trauma and social avoidance. Leave-one-out cross validation has been introduced to test the performance of Xgboost. We achieved AUC 0.95 and 0.61 AUPR. In addition, several case studies have been done to prove the accuracy of our result.