-
Notifications
You must be signed in to change notification settings - Fork 842
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 delegate to monitor training #847
Conversation
/azp run |
Azure Pipelines successfully started running 1 pipeline(s). |
Codecov Report
@@ Coverage Diff @@
## master #847 +/- ##
==========================================
- Coverage 85.17% 83.97% -1.21%
==========================================
Files 186 186
Lines 8588 8603 +15
Branches 512 508 -4
==========================================
- Hits 7315 7224 -91
- Misses 1273 1379 +106
Continue to review full report at Codecov.
|
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.
LGTM!
src/main/scala/com/microsoft/ml/spark/lightgbm/TrainUtils.scala
Outdated
Show resolved
Hide resolved
src/main/scala/com/microsoft/ml/spark/lightgbm/TrainUtils.scala
Outdated
Show resolved
Hide resolved
@@ -325,4 +325,11 @@ trait LightGBMParams extends Wrappable with DefaultParamsWritable with HasWeight | |||
|
|||
def getMinDataInLeaf: Int = $(minDataInLeaf) | |||
def setMinDataInLeaf(value: Int): this.type = set(minDataInLeaf, value) | |||
|
|||
var delegate: Option[LightGBMDelegate] = None |
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.
we should probably figure out how to make this inherit from param and expose in python layer, but maybe not for this PR
/azp run |
Azure Pipelines successfully started running 1 pipeline(s). |
It is needed to monitor and hook in each training iteration.
It's because how well training my model and how much better metrics such as auc in each boost round.
contacts: keunhyun.oh@ahnlab.com