Skip to main content
. 2022 Aug 8;12:13529. doi: 10.1038/s41598-022-17299-5

Table 2.

Model size, Training time and Inference speed per ML model family and input set.

Input features (num. features) Model size (MB) Training time (s) Inference speed (million pred./s)
Mean Max Mean Max Mean Min
Linear regression Input_orig. (4) 0.000657 0.000658 0.05 0.180189 6.64 6.227645
Input_hour/day (24) 0.000810 0.000811 1.64 3.120221 5.76 5.279960
Input_mix (44) 0.000966 0.000966 3.25 5.379071 5.12 4.735821
CART Input_orig. (4) 17.22 18.894751 0.53 0.549016 4.45 2.572063
Input_hour/day (24) 16.63 18.920263 2.85 3.013313 4.08 2.377553
Input_mix (44) 16.34 18.866186 5.26 5.626011 3.76 2.538655
Random-Forest input_orig. (4) 355.28 2047.384612 4.39 11.311513 0.65 0.284171
Input_hour/day (24) 354.10 2048.579925 24.76 61.137886 0.68 0.245368
Input_mix (44) 347.99 2043.802215 45.83 113.191902 0.68 0.294159