Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for scikit-learn 1.5 #125

Merged
merged 2 commits into from
May 23, 2024
Merged

Add support for scikit-learn 1.5 #125

merged 2 commits into from
May 23, 2024

Conversation

pavelzw
Copy link
Member

@pavelzw pavelzw commented May 23, 2024

No description provided.

@pavelzw pavelzw added the enhancement New feature or request label May 23, 2024
@pavelzw pavelzw merged commit 6d2d31e into main May 23, 2024
20 checks passed
@pavelzw pavelzw deleted the sklearn-15 branch May 23, 2024 19:12
Copy link

(benchmark 9213496979 / attempt 1)
Base results / Our results / Change

Model Size Dump Time Load Time
sklearn rf 20M 23.4 MiB / 3.0 MiB / 7.67 x 0.01 s / 0.03 s / 3.00 x 0.01 s / 0.03 s / 3.15 x
sklearn rf 20M lzma 6.5 MiB / 2.0 MiB / 3.22 x 9.91 s / 0.86 s / 0.09 x 0.40 s / 0.18 s / 0.45 x
sklearn rf 200M 238.8 MiB / 30.8 MiB / 7.75 x 0.07 s / 0.22 s / 3.13 x 0.14 s / 0.32 s / 2.37 x
sklearn rf 200M lzma 48.8 MiB / 14.8 MiB / 3.29 x 74.73 s / 14.22 s / 0.19 x 3.16 s / 1.16 s / 0.37 x
sklearn rf 1G 1302.2 MiB / 168.0 MiB / 7.75 x 0.38 s / 1.14 s / 3.01 x 0.79 s / 1.51 s / 1.91 x
sklearn rf 1G lzma 263.4 MiB / 99.1 MiB / 2.66 x 392.65 s / 86.15 s / 0.22 x 15.93 s / 6.77 s / 0.43 x
sklearn gb 2M 2.5 MiB / 1.2 MiB / 2.15 x 0.02 s / 0.21 s / 8.66 x 0.06 s / 0.25 s / 4.43 x
sklearn gb 2M lzma 0.6 MiB / 0.2 MiB / 3.77 x 0.78 s / 0.38 s / 0.48 x 0.07 s / 0.22 s / 3.31 x
lgbm gbdt 2M 2.6 MiB / 1.0 MiB / 2.78 x 0.04 s / 0.17 s / 4.07 x 0.01 s / 0.08 s / 9.16 x
lgbm gbdt 2M lzma 0.9 MiB / 0.5 MiB / 1.90 x 0.95 s / 0.35 s / 0.37 x 0.06 s / 0.11 s / 1.94 x
lgbm gbdt 5M 5.3 MiB / 1.9 MiB / 2.81 x 0.11 s / 0.33 s / 3.13 x 0.02 s / 0.17 s / 9.81 x
lgbm gbdt 5M lzma 1.7 MiB / 0.8 MiB / 1.96 x 2.48 s / 0.77 s / 0.31 x 0.11 s / 0.22 s / 2.08 x
lgbm gbdt 20M 22.7 MiB / 7.6 MiB / 3.00 x 0.35 s / 1.40 s / 4.02 x 0.07 s / 0.72 s / 10.23 x
lgbm gbdt 20M lzma 6.3 MiB / 3.0 MiB / 2.09 x 13.84 s / 3.50 s / 0.25 x 0.41 s / 0.89 s / 2.17 x
lgbm gbdt 100M 101.1 MiB / 33.0 MiB / 3.06 x 1.52 s / 6.34 s / 4.19 x 0.29 s / 44.68 s / 152.66 x
lgbm gbdt 100M lzma 25.6 MiB / 10.6 MiB / 2.41 x 64.96 s / 17.69 s / 0.27 x 1.64 s / 3.76 s / 2.29 x
lgbm rf 10M 10.9 MiB / 3.2 MiB / 3.46 x 0.15 s / 0.40 s / 2.73 x 0.03 s / 0.28 s / 9.52 x
lgbm rf 10M lzma 0.7 MiB / 0.4 MiB / 1.86 x 1.23 s / 0.58 s / 0.47 x 0.08 s / 0.29 s / 3.54 x

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant