Skip to content

Conversation

@jmenblaz
Copy link

Current version of the function partial doesn't work with Random Intercept BART models. Checking the code of the function the step pd <- pdbart(model, xind = x.vars, levs = lev, pl=FALSE) use the model object (bart, rbart...) itself. However, using rbart as an input of the step return error: x.train must be a matrix, data.frame, formula, fitted bart model, or dbartsSampler.

Changing the input object, fitobj (created previously) for model object works properly for both, bart and rbart models.

I will check the same for other function (spartials(), etc) and make the PR then

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant