Table 1. AUC mean (and standard deviation) determined by 10 repetitions of 10-fold cross-validation.
Algorithm | Sharing Food | Creativity | Average |
---|---|---|---|
ET (5000 trees) | 0.785 (σ = 0.016) | 0.893 (σ = 0.01) | 0.881 (σ = 0.01) |
RF (5000 trees) | 0.802 (σ = 0.017) | 0.892 (σ = 0.006) | 0.885 (σ = 0.006) |
RGF (500 trees, L2 = sL2 = 1.0, square loss) | 0.835 (σ = 0.017) | 0.921 (σ = 0.012) | 0.927 (σ = 0.011) |
RGF2 | 0.848 (σ = 0.025) | 0.926 (σ = 0.013) | 0.932 (σ = 0.016) |
The last column (Average) denotes AUC obtained by taking a mean of predictions of both games for each child.