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

feat: Add support for loading / writing from files #134

Merged
merged 3 commits into from
Aug 23, 2024
Merged

feat: Add support for loading / writing from files #134

merged 3 commits into from
Aug 23, 2024

Conversation

0xbe7a
Copy link
Contributor

@0xbe7a 0xbe7a commented Aug 23, 2024

No description provided.

@0xbe7a 0xbe7a requested a review from pavelzw as a code owner August 23, 2024 09:11
slim_trees/sklearn_tree.py Outdated Show resolved Hide resolved
slim_trees/sklearn_tree.py Outdated Show resolved Hide resolved
@0xbe7a 0xbe7a merged commit 9461551 into main Aug 23, 2024
20 checks passed
@0xbe7a 0xbe7a deleted the open-file branch August 23, 2024 09:25
@pavelzw pavelzw added the enhancement New feature or request label Aug 23, 2024
Copy link

github-actions bot commented Aug 23, 2024

(benchmark 10523297859 / 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 / 2.34 x 0.02 s / 0.03 s / 1.47 x
sklearn rf 20M lzma 6.5 MiB / 2.0 MiB / 3.21 x 9.67 s / 0.86 s / 0.09 x 0.41 s / 0.15 s / 0.37 x
sklearn rf 200M 238.8 MiB / 30.8 MiB / 7.75 x 0.15 s / 0.23 s / 1.57 x 0.23 s / 0.38 s / 1.65 x
sklearn rf 200M lzma 48.8 MiB / 14.8 MiB / 3.29 x 72.18 s / 14.25 s / 0.20 x 3.18 s / 1.14 s / 0.36 x
sklearn rf 1G 1302.2 MiB / 168.0 MiB / 7.75 x 0.73 s / 1.14 s / 1.55 x 1.16 s / 1.87 s / 1.61 x
sklearn rf 1G lzma 262.4 MiB / 99.1 MiB / 2.65 x 372.80 s / 79.86 s / 0.21 x 16.11 s / 6.80 s / 0.42 x
sklearn gb 2M 2.5 MiB / 1.1 MiB / 2.16 x 0.02 s / 0.22 s / 8.89 x 0.03 s / 0.21 s / 7.43 x
sklearn gb 2M lzma 0.6 MiB / 0.2 MiB / 3.76 x 0.76 s / 0.36 s / 0.48 x 0.07 s / 0.23 s / 3.35 x
lgbm gbdt 2M 2.6 MiB / 1.0 MiB / 2.78 x 0.05 s / 0.18 s / 3.90 x 0.01 s / 0.12 s / 14.09 x
lgbm gbdt 2M lzma 0.9 MiB / 0.5 MiB / 1.90 x 0.89 s / 0.36 s / 0.40 x 0.06 s / 0.11 s / 1.98 x
lgbm gbdt 5M 5.3 MiB / 1.9 MiB / 2.81 x 0.09 s / 0.35 s / 3.77 x 0.02 s / 0.17 s / 10.20 x
lgbm gbdt 5M lzma 1.7 MiB / 0.8 MiB / 1.96 x 2.15 s / 0.73 s / 0.34 x 0.11 s / 0.22 s / 2.12 x
lgbm gbdt 20M 22.7 MiB / 7.6 MiB / 3.00 x 0.38 s / 1.45 s / 3.79 x 0.07 s / 0.75 s / 10.47 x
lgbm gbdt 20M lzma 6.3 MiB / 3.0 MiB / 2.09 x 12.10 s / 3.44 s / 0.28 x 0.41 s / 0.88 s / 2.16 x
lgbm gbdt 100M 101.1 MiB / 33.0 MiB / 3.06 x 1.78 s / 6.63 s / 3.73 x 0.36 s / 56.62 s / 157.80 x
lgbm gbdt 100M lzma 25.6 MiB / 10.6 MiB / 2.41 x 57.84 s / 16.89 s / 0.29 x 1.67 s / 3.74 s / 2.24 x
lgbm rf 10M 10.9 MiB / 3.2 MiB / 3.46 x 0.17 s / 0.42 s / 2.42 x 0.03 s / 0.27 s / 9.81 x
lgbm rf 10M lzma 0.7 MiB / 0.4 MiB / 1.86 x 1.22 s / 0.60 s / 0.49 x 0.08 s / 0.30 s / 3.72 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.

2 participants