Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
19562a90
Unverified
Commit
19562a90
authored
Mar 22, 2018
by
Guolin Ke
Committed by
GitHub
Mar 22, 2018
Browse files
[R-package] add missing objectives
parent
bcd77003
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
R-package/R/utils.R
R-package/R/utils.R
+6
-1
No files found.
R-package/R/utils.R
View file @
19562a90
...
...
@@ -160,7 +160,12 @@ lgb.check.params <- function(params) {
lgb.check.obj
<-
function
(
params
,
obj
)
{
# List known objectives in a vector
OBJECTIVES
<-
c
(
"regression"
,
"regression_l1"
,
"regression_l2"
,
"huber"
,
"fair"
,
"poisson"
,
"binary"
,
"lambdarank"
,
"multiclass"
)
OBJECTIVES
<-
c
(
"regression"
,
"regression_l1"
,
"regression_l2"
,
"mean_squared_error"
,
"mse"
,
"l2_root"
,
"root_mean_squared_error"
,
"rmse"
,
"mean_absolute_error"
,
"mae"
,
"quantile"
,
"huber"
,
"fair"
,
"poisson"
,
"binary"
,
"lambdarank"
,
"multiclass"
,
"softmax"
,
"multiclassova"
,
"multiclass_ova"
,
"ova"
,
"ovr"
,
"xentropy"
,
"cross_entropy"
,
"xentlambda"
,
"cross_entropy_lambda"
,
"mean_absolute_percentage_error"
,
"mape"
,
"gamma"
,
"tweedie"
)
# Check whether the objective is empty or not, and take it from params if needed
if
(
!
is.null
(
obj
))
{
params
$
objective
<-
obj
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment