Table 3 Hyperparameters and their tuning ranges.
From: Machine learning models based wear performance prediction of AZ31/TiC composites
Model | Hyperparameter | Values |
|---|---|---|
Decision Tree | max_depth | 3, 5, 10, None |
min_samples_split | 2, 5, 10 | |
Random Forest | n_estimators | 50, 100, 200 |
max_depth | 5, 10, None | |
min_samples_split | 2, 5, 10 | |
Gradient Boost | n_estimators | 50, 100, 200 |
learning_rate | 0.01, 0.1, 0.2 | |
max_depth | 3, 5, 10 | |
XGBoost | n_estimators | 50, 100, 200 |
max_depth | 3, 5, 10 | |
learning_rate | 0.01, 0.1, 0.2 |
