BrainSort: a Machine Learning Toolkit for Brain Connectome Data Analysis and Visualization

Abstract

In recent years, applying machine learning methods to neurological and psychiatric disorder diagnoses has grasped the interest of many researchers; however, currently available machine learning toolboxes usually require somewhat intermediate programming knowledge. In order to use machine learning methods more quickly and conveniently, we developed an intuitive toolbox named BrainSort. BrainSort used Python as the main programming languages and employed a hospitable Graphical User Interface (GUI). The toolbox is user-friendly for researchers and clinical doctors with little to no prior programming skills. It enables the client to choose from multiple machine learning methods, such as support vector machine (SVM), k-nearest neighbors (k-NN), and convolutional neural network (CNN) for data processing and training. Using BrainSort, doctors and researchers can calculate and visualize the correlation between brain connectome topology parameters and the symptom in question without prolonged programming training, empowering them to use the powerful tool of machine learning in their studies and practices.