-
Notifications
You must be signed in to change notification settings - Fork 629
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Training ranger model fails when setting write.forest #1177
Comments
It's |
Thanks @mnwright for pointing out my mistake here. I've corrected it and also noticed that when using caret I have to use `caret::predict.train when packaging the model with carrier. This fixes the packaging when using ranger along with caret. The revised code works unless I specify
So the minor bug is that the user can not set Corrected code:
|
The initial issue was that you used
This code: rfSimple = train(
Species ~ .,
data = iris,
method = "ranger",
num.trees = 5,
write.forest=TRUE
) failed because
|
@topepo naiive question but how do we know that |
When building a simple
ranger
model for iris (which is the same as in the official docs, caret fails towrite.forrest=T
ranger
), and thus the resulting model can be properly persisted (e.g. viacarrier
for model serving viamlflow
)Session Info:
FYI @mnwright
The text was updated successfully, but these errors were encountered: