HGB build scalability comparison

Hardware

CPU

Intel(R) Core(TM) Ultra X7 358H

X86_64, 16 physical cores, 16 logical CPUs

31 GB RAM

GPU(s)

  • level_zero:gpu:0: Intel(R) Arc(TM) B390 GPU (29 GB)

XS (1,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

S (1,000 x 100)

n_iter=100, max_leaf_nodes=31, max_features=0.5

S-thin (5,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

M (20,000 x 50)

n_iter=100, max_leaf_nodes=31, max_features=0.5

amazon_employee_access (26,215 x 9)

n_iter=150, max_leaf_nodes=31, 9 categorical, min_samples_leaf=20

kddcup09_churn (40,000 x 207)

n_iter=150, max_leaf_nodes=15, 34 categorical, min_samples_leaf=50

M-very-thin (50,000 x 5)

n_iter=100, max_leaf_nodes=31, max_features=1.0

L-leaf255 (100,000 x 100)

n_iter=20, max_leaf_nodes=255, max_features=0.5

L (100,000 x 100)

n_iter=100, max_leaf_nodes=31, max_features=0.5

M-stumps (200,000 x 100)

n_iter=100, max_leaf_nodes=2, max_features=0.5

year_prediction_msd (463,810 x 90)

n_iter=200, max_leaf_nodes=127, min_samples_leaf=5

covtype (464,809 x 12)

n_iter=200, max_leaf_nodes=127, 2 categorical, min_samples_leaf=5

L-thin (500,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

L-stumps (1,000,000 x 100)

n_iter=100, max_leaf_nodes=2, max_features=0.5

susy (4,500,000 x 18)

n_iter=200, max_leaf_nodes=127, min_samples_leaf=5

Hardware

CPU

Intel(R) Xeon(R) 6787P

X86_64, 172 physical cores, 344 logical CPUs

499 GB RAM

GPU(s)

No GPU detected.

XS (1,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

S (1,000 x 100)

n_iter=100, max_leaf_nodes=31, max_features=0.5

S-thin (5,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

M (20,000 x 50)

n_iter=100, max_leaf_nodes=31, max_features=0.5

amazon_employee_access (26,215 x 9)

n_iter=150, max_leaf_nodes=31, 9 categorical, min_samples_leaf=20

kddcup09_churn (40,000 x 207)

n_iter=150, max_leaf_nodes=15, 34 categorical, min_samples_leaf=50

M-very-thin (50,000 x 5)

n_iter=100, max_leaf_nodes=31, max_features=1.0

L-leaf255 (100,000 x 100)

n_iter=20, max_leaf_nodes=255, max_features=0.5

L (100,000 x 100)

n_iter=100, max_leaf_nodes=31, max_features=0.5

M-stumps (200,000 x 100)

n_iter=100, max_leaf_nodes=2, max_features=0.5

year_prediction_msd (463,810 x 90)

n_iter=200, max_leaf_nodes=127, min_samples_leaf=5

covtype (464,809 x 12)

n_iter=200, max_leaf_nodes=127, 2 categorical, min_samples_leaf=5

L-thin (500,000 x 10)

n_iter=100, max_leaf_nodes=31, max_features=1.0

L-stumps (1,000,000 x 100)

n_iter=100, max_leaf_nodes=2, max_features=0.5

susy (4,500,000 x 18)

n_iter=200, max_leaf_nodes=127, min_samples_leaf=5