Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add variable type checks

Browse Source
This commit is contained in:
El Potaeto 2014-12-29 10:31:17 +01:00
parent 6b96737811
commit 755be4b846
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
R-package/R/xgb.importance.R
View File

@ -29,6 +29,12 @@
#' #'
#' @export #' @export
xgb.importance <- function(feature_names, filename_dump){ xgb.importance <- function(feature_names, filename_dump){
if (class(feature_names) != "character") {
stop("feature_names: Has to be a vector of character. See help to see where to get it.")
}
if (class(filename_dump) != "character" & file.exists(filename_dump)) {
stop("filename_dump: Has to be a path to the model dump file.")
}
text <- readLines(filename_dump) text <- readLines(filename_dump)
if(text[2] == "bias:"){ if(text[2] == "bias:"){
result <- linearDump(feature_names, text) result <- linearDump(feature_names, text)