dj_ski_mask
dj_ski_mask

Reputation: 65

H2O R Variable Importance Truncated List

I have a data set with over 400 features that I am estimating with GBM using H2O atop R. When I use the variable importance function (h2o.varimp) it only shows me the head and tail of the full ranked variable list. Is there a way to have the entire list displayed?

Upvotes: 4

Views: 2764

Answers (2)

Erin LeDell
Erin LeDell

Reputation: 8819

This is not specific to variable importance, this is just how H2O displays H2O Frames in the R console. If you want to view the whole frame, you could convert it to an R data.frame and then print it.

df <- as.data.frame(h2o.varimp(model))
print(df)

Upvotes: 9

tylers
tylers

Reputation: 1

(summary) will show all stats with extraction from h2o.varimp. Then save the table of variable importance

mymodel <- summary(model)

write.table(mymodel, file = "mymodel.txt", sep = "\t", quote = FALSE, row.names = TRUE)

Upvotes: 0

Related Questions