-
Notifications
You must be signed in to change notification settings - Fork 129
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
increase the default epochs to 1000 for linear regression #394
Conversation
Signed-off-by: Xun Zhang <xunzh@amazon.com>
@@ -58,7 +58,7 @@ public class LinearRegression implements Trainable, Predictable { | |||
//RMSProp | |||
private static final double DEFAULT_DECAY_RATE = 0.9; | |||
|
|||
private static final int DEFAULT_EPOCHS = 10; | |||
private static final int DEFAULT_EPOCHS = 1000; | |||
private static final int DEFAULT_INTERVAL = -1; | |||
private static final int DEFAULT_BATCH_SIZE = 1; |
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.
Can you help fix the hard coded issue in line 203
LinearSGDTrainer linearSGDTrainer = new LinearSGDTrainer(objective, optimiser, epochs, DEFAULT_INTERVAL, DEFAULT_BATCH_SIZE, seed);
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.
Sure, do you mean DEFAULT_INTERVAL, DEFAULT_BATCH_SIZE, seed these 3 hard coded ones?
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.
yes
Signed-off-by: Xun Zhang <xunzh@amazon.com>
Codecov Report
@@ Coverage Diff @@
## main #394 +/- ##
=========================================
Coverage 91.43% 91.43%
Complexity 528 528
=========================================
Files 59 59
Lines 1529 1529
Branches 122 122
=========================================
Hits 1398 1398
Misses 89 89
Partials 42 42
Flags with carried forward coverage won't be shown. Click here to find out more. Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
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. Thanks for the change!
* increase the default epochs to 1000 for linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> * make more parameters configuration in linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> (cherry picked from commit b760180)
* increase the default epochs to 1000 for linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> * make more parameters configuration in linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> (cherry picked from commit b760180) Co-authored-by: Xun Zhang <xunzh@amazon.com>
* increase the default epochs to 1000 for linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> * make more parameters configuration in linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> (cherry picked from commit b760180)
* increase the default epochs to 1000 for linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> * make more parameters configuration in linear regression Signed-off-by: Xun Zhang <xunzh@amazon.com> (cherry picked from commit b760180) Co-authored-by: Xun Zhang <xunzh@amazon.com>
Signed-off-by: Xun Zhang xunzh@amazon.com
Description
Increase the number of iteration for epochs to improve convergence of linear regression.
Issues Resolved
Check List
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
For more information on following Developer Certificate of Origin and signing off your commits, please check here.