Skip to main content

Table 3 Classification models’ performance. Machine learning models performance (LR = Logistic regression, RF = Random Forest, XGB = XGBoost.) evaluated with the Accuracy, F1 score, Precision, Recall and Area Under the Receiver Operating Characteristic Curve (AUROC)

From: Inferring psychological traits from spending categories and dynamic consumption patterns

Individual traits

Model

Accuracy

Precision

Recall

F1 measure

AUROC

Mean

Std

Mean

Std

Mean

Std

Mean

Std

Mean

Std

Materialism

LR

0.377

0.032

0.375

0.034

0.378

0.032

0.371

0.033

0.544

0.020

RF

0.423

0.026

0.420

0.028

0.423

0.026

0.420

0.026

0.588

0.019

XGB

0.409

0.027

0.406

0.027

0.409

0.027

0.406

0.028

0.580

0.016

Extraversion

LR

0.385

0.029

0.362

0.050

0.386

0.029

0.362

0.039

0.567

0.027

RF

0.386

0.026

0.384

0.025

0.387

0.026

0.383

0.025

0.574

0.020

XGB

0.398

0.033

0.397

0.032

0.398

0.033

0.396

0.032

0.573

0.029

Agreeableness

LR

0.345

0.026

0.333

0.045

0.345

0.026

0.321

0.035

0.508

0.023

RF

0.377

0.019

0.375

0.020

0.377

0.019

0.374

0.020

0.547

0.018

XGB

0.363

0.028

0.363

0.030

0.364

0.028

0.360

0.029

0.540

0.026

Conscientiousness

LR

0.361

0.023

0.361

0.026

0.362

0.023

0.360

0.025

0.533

0.010

RF

0.361

0.036

0.362

0.037

0.361

0.036

0.360

0.036

0.529

0.027

XGB

0.345

0.022

0.345

0.021

0.345

0.022

0.344

0.021

0.512

0.018

Neuroticism

LR

0.392

0.026

0.393

0.027

0.392

0.026

0.386

0.026

0.570

0.029

RF

0.401

0.018

0.402

0.019

0.401

0.018

0.399

0.019

0.558

0.015

XGB

0.377

0.015

0.376

0.016

0.377

0.015

0.374

0.014

0.553

0.013

Openness to experience

LR

0.344

0.014

0.338

0.049

0.344

0.013

0.322

0.036

0.512

0.018

RF

0.348

0.034

0.346

0.034

0.348

0.034

0.345

0.034

0.520

0.024

XGB

0.331

0.026

0.329

0.025

0.331

0.026

0.324

0.028

0.508

0.013

Self-control

LR

0.410

0.026

0.408

0.026

0.411

0.026

0.407

0.026

0.585

0.023

RF

0.394

0.026

0.392

0.027

0.395

0.026

0.391

0.026

0.575

0.029

XGB

0.379

0.034

0.377

0.035

0.379

0.034

0.376

0.035

0.564

0.033