fix base_score problem, tidymodels#65

JiaxiangBU · JiaxiangBU · commit a48c7d872ca7 · 2019-10-20T15:13:40.000+08:00
diff --git a/R/model-xgboost.R b/R/model-xgboost.R
@@ -140,7 +140,7 @@ build_fit_formula_xgb <- function(parsedmodel) {
     assigned <- 1
   } else if (objective %in% c("binary:logistic", "reg:logistic")) {
     assigned <- 1
-    f <- expr(1 - 1 / (1 + exp(!!f)))
+    f <- expr(1 - 1 / (1 + exp(!!f + log(!!base_score/(1-!!base_score)))))
   }
   if (assigned == 0) {
     stop("Only objectives 'binary:logistic', 'reg:linear', 'reg:logistic', 'binary:logitraw' are supported yet.")

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,7 @@ build_fit_formula_xgb <- function(parsedmodel) {`
`140`	`140`	`assigned <- 1`
`141`	`141`	`} else if (objective %in% c("binary:logistic", "reg:logistic")) {`
`142`	`142`	`assigned <- 1`
`143`		`- f <- expr(1 - 1 / (1 + exp(!!f)))`
	`143`	`+ f <- expr(1 - 1 / (1 + exp(!!f + log(!!base_score/(1-!!base_score)))))`
`144`	`144`	`}`
`145`	`145`	`if (assigned == 0) {`
`146`	`146`	`stop("Only objectives 'binary:logistic', 'reg:linear', 'reg:logistic', 'binary:logitraw' are supported yet.")`