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
Enhance class SoftmaxRegression #456
Conversation
2 : add function Train 3 : fix bug--did not initialize fitIntercept
inputSize{inputSize}, | ||
numClasses{numClasses}, | ||
lambda{0.0001}, | ||
fitIntercept{fitIntercept} |
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.
For consistency, do you mind if I change this to ()
initialization vs. {}
initialization?
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.
No problem(I use {} to avoid narrow conversion)
Looks good to me; I made a few comments. Let me know what you think. I'll add a test for |
name : Tham Ngap Wei Thanks for adding the unit test. |
Enhance class SoftmaxRegression
Okay, I made some simple changes in cbeb3ea..f85a9b2; let me know if you see anything wrong. If you are interested, here are some future possibilities for refactoring/improving the SoftmaxRegression code further:
Don't feel obligated to do any of these -- I'm just tossing them out there as ideas. :) Thanks for your contribution! :) |
Your suggestions are good, about
Some(or many) classes of mlpack always ask for the inputSize(feature size) from the users, I think they could be removed too, because many algorithms should be able to find out the inputSize from the training data
|
1 : add function Serialize to class SoftmaxRegression
2 : add function Train to class SoftmaxRegression
3 : fix bug--did not initialize fitIntercept
The test case "SoftmaxRegressionFunctionEvaluate" always fail on my machine