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

Support saving/loading of GBDT models #269

Merged
merged 19 commits into from
Dec 8, 2023

Conversation

xnuohz
Copy link
Contributor

@xnuohz xnuohz commented Dec 5, 2023

Copy link

codecov bot commented Dec 5, 2023

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (13d9f35) 93.20% compared to head (7e1622f) 93.25%.

Additional details and impacted files
@@            Coverage Diff             @@
##           master     #269      +/-   ##
==========================================
+ Coverage   93.20%   93.25%   +0.04%     
==========================================
  Files         115      115              
  Lines        5680     5719      +39     
==========================================
+ Hits         5294     5333      +39     
  Misses        386      386              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Copy link
Contributor

@weihua916 weihua916 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! Can you please add tests for loading/saving? You can follow https://github.com/pyg-team/pytorch-frame/blob/master/test/utils/test_io.py#L14-L16

@xnuohz
Copy link
Contributor Author

xnuohz commented Dec 6, 2023

@weihua916 Add a test case for it. Is it correct?

Copy link
Contributor

@weihua916 weihua916 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! Left some comments to simplify and strengthen the tests.

test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
Copy link
Contributor

@weihua916 weihua916 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks mostly good! Final nit.

test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
examples/tuned_gbdt.py Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
@xnuohz
Copy link
Contributor Author

xnuohz commented Dec 7, 2023

@weihua916 @yiweny Done. (Bad network, took some time to install mypy lol)

Copy link
Contributor

@yiweny yiweny left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM after resolving the comment. Let's merge after a pass from @weihua916

CHANGELOG.md Outdated Show resolved Hide resolved
test/gbdt/test_gbdt.py Outdated Show resolved Hide resolved
Copy link
Contributor

@weihua916 weihua916 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@weihua916 weihua916 merged commit 4ef20f4 into pyg-team:master Dec 8, 2023
11 checks passed
@xnuohz xnuohz deleted the gbdts/save-and-load branch December 8, 2023 07:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants