Model | Hyperparameters for Exp. 1A and 1B | Hyperparameters for Exp. 2A and 2B |
---|---|---|
Support Vector Machine | C = 1, kernel = ‘rbf’, gamma = ‘scale’ | C = 1, kernel = ‘rbf’, gamma = '0.1’ |
Random Forest | Nb_estimators = 600, criterion = “gini”, max_depth = 15 | Nb_estimators = 300, criterion = “mse”, max_depth = 25 |
XGBoost | Nb_estimators = 100, gamma = 0, max_depth = 6, learning_rate = 0.3, reg_lambda = 1 | Nb_estimators = 100, gamma = 0, max_depth = 6, learning_rate = 0.3, reg_lambda = 1 |
CatBoost | Iterations = 1000, depth = 6, learning_rate = 0.08 | Iterations = 1000, depth = 6, learning_rate = 0.04 |