Update utils.R

This commit is contained in:
Tong He 2015-03-20 22:41:47 -07:00
parent 7ced224722
commit 5648bec8a3

View File

@ -193,7 +193,8 @@ xgb.iter.eval <- function(booster, watchlist, iter, feval = NULL, prediction = F
if (length(names(w)) == 0) { if (length(names(w)) == 0) {
stop("xgb.eval: name tag must be presented for every elements in watchlist") stop("xgb.eval: name tag must be presented for every elements in watchlist")
} }
ret <- feval(predict(booster, w[[1]]), w[[1]]) preds <- predict(booster, w[[1]])
ret <- feval(preds, w[[1]])
msg <- paste(msg, "\t", names(w), "-", ret$metric, ":", ret$value, sep="") msg <- paste(msg, "\t", names(w), "-", ret$metric, ":", ret$value, sep="")
} }
} }