-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Adds the ability to load a pre-trained LightGBM file and import it into ML.Net. #6569
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
Conversation
/// <param name="catalog">The <see cref="RegressionCatalog"/>.</param> | ||
/// <param name="lightGbmModel"> A pre-trained <see cref="System.IO.Stream"/> of a LightGBM model file inferencing</param> | ||
/// <param name="featureColumnName">The name of the feature column. The column data must be a known-sized vector of <see cref="System.Single"/>.</param> | ||
public static LightGbmRegressionTrainer LightGbm(this RegressionCatalog.RegressionTrainers catalog, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What happen if loading an LightGBM model newer than 2.3.1, maybe we can add test for that scenario?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So at least "officially" we aren't going to support it. But @torronen said he would test it and let us know since the model format itself should be the same. Since we are going to be updating to the latest version though this year this is only a temporary situation.
Codecov Report
Additional details and impacted files@@ Coverage Diff @@
## main #6569 +/- ##
==========================================
+ Coverage 68.39% 68.78% +0.38%
==========================================
Files 1174 1215 +41
Lines 248045 251729 +3684
Branches 25909 26250 +341
==========================================
+ Hits 169649 173140 +3491
- Misses 71627 71774 +147
- Partials 6769 6815 +46
Flags with carried forward coverage won't be shown. Click here to find out more.
|
Merging, known issue #6686. |
Fixes #6296.
Adds the ability to load a pre-trained LightGBM file and import it into ML.Net.