A decision support system for selecting the most suitable machine learning in healthcare using user parameters and requirements
Yashodhan Ketkar, Sushopti Gawade
Abstract
The application of machine learning in the medical field is still limited. The main reason behind the lack of use is the unavailability of an easy-to-use machine learning system that targets non-technical users. The objective of this paper is to propose an automated machine learning system to aid non-technical users. The proposed system provides the user with simple choices to provide suggestions to the system. The system uses the combination of the user’s choices and performance evaluation to select the most suited model from available options. In this study, we employed the system on a Parkinson’s disease dataset. The templates for support vector machine and random forest algorithms are provided to the system. Support vector machines and random forests were able to produce 80% and 75% accuracy, respectively. The system used performance parameters of the system and user choices to select the most suited models for each test case. The support vector machine was selected as the most suited model in three test cases, while random forest was selected as the most suited for one test case. The test cases also showed that the weighted time parameter impacted the results heavily.