Skip to content

Instantly share code, notes, and snippets.

What would you like to do?
# feature importance plot
bst <- xgb.train(data=dtrain, max.depth=5.5, eta=.02, nthread = 2, nround=5, watchlist=watchlist, objective = "binary:logistic")
xgb.dump(bst, with.stats = T)
xgb.plot.tree(model = bst)
importance_matrix <- xgb.importance(model = bst)
xgb.plot.importance(importance_matrix = importance_matrix)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.