New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Potential Fix For VarImp To Work With Categorical Models Such as Method pda #1355

Open

thomashigginsdatascientist opened this issue Feb 10, 2024 · 0 comments

thomashigginsdatascientist commented Feb 10, 2024

insert on line 134 filterVarImp:

y_unlist = unlist(y)

change line 135 filterVarImp:

outStat <- apply(x, 2, testFunc, y = y_unlist)

insert on line 23 of varImp.train:

y_numeric <- is.numeric(y_dat)
if(y_numeric){
}else{
y_dat$.outcome <- as.factor(y_dat$.outcome)
y_dat$.outcome <- as.numeric(y_dat$.outcome)
}

# for free to join this conversation on GitHub. Already have an account? # to comment