zmjones / edarf

exploratory data analysis using random forests
MIT License
68 stars 11 forks source link

class probability bug #20

Closed zmjones closed 9 years ago

zmjones commented 9 years ago

error when running the iris example with class probability output

library(randomForest)
library(edarf)
data(iris)
fit <- randomForest(Species ~ ., iris)
pd <- partial_dependence(fit, iris, "Petal.Width", type = "prob")
plot(pd)
Error in as.character(x$label) : 
  cannot coerce type 'closure' to vector of type 'character'