Table 2 Parameter setting of these models.
model | Parameter name | Parameter range | Best parameter |
---|---|---|---|
IPM-SVM | C | [0.1, 1, 10, 100] | 100 |
gamma | [0.001, 0.01, 0.1, 1] | 1 | |
kernel | [‘rbf’, ‘linear’, ‘poly’] | ‘rbf’ | |
IPM-DTC | max_depth | [1,2,3,4,5,6,7,8,9,10] | 10 |
IPM-KNN | n_neighbors | – | 4 |