[ML] Add loss_function to regression#56118
[ML] Add loss_function to regression#56118dimitris-athanasiou merged 4 commits intoelastic:masterfrom
Conversation
Adds parameters `loss_function` and `loss_function_parameter` to regression.
|
Pinging @elastic/ml-core (:ml) |
benwtrent
left a comment
There was a problem hiding this comment.
I have an overall design question.
To my knowledge, only the huber function will actually use the lossFunctionParameter. It almost seems to me that this could be a named object.
If we don't want the complexity a named object requires, what do you think of adding validations? We shouldn't allow the lossFunctionParameter unless it is being used.
|
|
|
@benwtrent you are right, in a canonical form MSLE does not require a parameter, however, @tveasey had a useful insight that |
|
|
||
| `loss_function`:::: | ||
| (Optional, string) | ||
| The loss function used during regression. Available options are `mse` (Mean Squared Error), |
There was a problem hiding this comment.
I don't think this requires capitalization
| The loss function used during regression. Available options are `mse` (Mean Squared Error), | |
| The loss function used during regression. Available options are `mse` (mean squared error), |
| `loss_function`:::: | ||
| (Optional, string) | ||
| The loss function used during regression. Available options are `mse` (Mean Squared Error), | ||
| `msle` (Mean Squared Logarithmic Error), `huber` (Pseudo-Huber loss). Defaults to `mse`. |
There was a problem hiding this comment.
Ditto re mean squared logarithmic, whereas huber seems to be a person's name and can be capitalized
| `msle` (Mean Squared Logarithmic Error), `huber` (Pseudo-Huber loss). Defaults to `mse`. | |
| `msle` (mean squared logarithmic error), `huber` (Pseudo-Huber loss). Defaults to `mse`. |
|
|
||
| `loss_function_parameter`:::: | ||
| (Optional, double) | ||
| A strictly positive number that is used as a parameter to the `loss_function`. |
There was a problem hiding this comment.
A strictly positive number ...
Why is "strictly" required here?
There was a problem hiding this comment.
Fair point! I'll remove it.
|
@elasticmachine update branch |
Until elastic#56118 is backported to 7.x.
Until #56118 is backported to 7.x.
Adds parameters `loss_function` and `loss_function_parameter` to regression. Backport of elastic#56118
Adds parameters
loss_functionandloss_function_parameterto regression.