Skip to main content
. 2016 Feb 9;6(2):e732. doi: 10.1038/tp.2015.221

Table 3. Accuracy of machine learning algorithms.

Algorithm AUC Features used
Decision tree 0.933 2/65
Random forest 0.952 9/65
Support Vector Classification 0.965 5/65
Logistic Regression 0.962 5/65
Categorical lasso 0.962 5/65
Linear discriminant analysis 0.964 5/65

Using mutual information feature selection methods. Categorical Lasso was implemented as a Logistic Regression model with l1 regularization and our Logistic Regression model utilized l2 regularization. Support Vector Classification was applied using the linear kernel.