Binary classification (low vs. high) with 2,181 1-min segments of ['gsr', 'bpm', 'rri', 'temp']
.
$ python baseline.py -r ~/data/dailyLife2/datasets/60s -e ~/data/dailyLife2/metadata/esms_activity.csv -s 1 -t arousal --cv 'kfold' --splits 5 --shuffle --gpu
Metric |
Random |
Majority |
Class ratio |
Gaussian NB |
XGBoost |
acc. |
0.501611 |
0.52132 |
0.499783 |
0.553404 |
0.55662 |
auroc |
0.5 |
0.5 |
0.499221 |
0.577678 |
0.586021 |
bacc. |
0.501703 |
0.5 |
0.499221 |
0.557775 |
0.555669 |
f1 |
0.510292 |
0.685352 |
0.515729 |
0.513229 |
0.575862 |
$ python baseline.py -r ~/data/dailyLife2/datasets/60s -e ~/data/dailyLife2/metadata/esms_activity.csv -s 1 -t valence --cv 'kfold' --splits 5 --shuffle --gpu
Metric |
Random |
Majority |
Class ratio |
Gaussian NB |
XGBoost |
acc. |
0.491061 |
0.786795 |
0.661163 |
0.746965 |
0.75929 |
auroc |
0.5 |
0.5 |
0.498554 |
0.521622 |
0.568744 |
bacc. |
0.48922 |
0.5 |
0.498554 |
0.510779 |
0.521717 |
f1 |
0.603419 |
0.880677 |
0.783953 |
0.847594 |
0.859448 |