Open Access

Table 3

Best set of values for the XGB hyperparameters after performing a grid search.

Iteration 1 Iteration 2


Hyperparameter DS0 DS1 DS2 DS3 DS4 DS5 DS6 DS7 DS0 DS1 DS2 DS3 DS4 DS5 DS6 DS7
colsample_bytree 2/3 1/2 1/3 1/2 1/2 1/2 1/2 1/2 3/4 2/5 2/5 1/2 1/2 2/5 2/5 1/2
eta 0.04 0.02 0.02 0.03 0.02 0.02 0.02 0.02 0.03 0.02 0.02 0.02 0.02 0.02 0.02 0.02
gamma 1 10 3 1 5 5 1 1 1 5 3 3 3 0 1 5
max_depth 7 7 7 7 7 7 7 7 9 7 8 9 8 9 9 9
min_child_weight 20 20 20 20 20 20 10 10 15 25 25 25 15 15 10 10
subsample 1.0 1.0 0.90 0.95 0.90 0.90 0.95 0.95 1.0 1.0 1.0 1.0 0.90 0.90 0.90 0.90
nrounds 359 1422 1453 877 1249 881 1275 1153 470 1373 1516 1285 958 1078 1114 1089

Notes. The nrounds parameter indicates the number of iterations that the XGB algorithm should perform to obtain the best model using the optimal values of the hyperparameters for the given data set.

Current usage metrics show cumulative count of Article Views (full-text article views including HTML views, PDF and ePub downloads, according to the available data) and Abstracts Views on Vision4Press platform.

Data correspond to usage on the plateform after 2015. The current usage metrics is available 48-96 hours after online publication and is updated daily on week days.

Initial download of the metrics may take a while.