| gbt.complexity {agtboost} | R Documentation |
gbt.complexity creates a list of hyperparameters from a model
gbt.complexity(model, type)
model |
object or pointer to object of class |
type |
currently supports "xgboost" or "lightgbm" |
Returns the complexity of model in terms of hyperparameters associated
to model type.
list with type hyperparameters.
set.seed(123) library(agtboost) n <- 10000 xtr <- as.matrix(runif(n, 0, 4)) ytr <- rnorm(n, xtr, 1) xte <- as.matrix(runif(n, 0, 4)) yte <- rnorm(n, xte, 1) model <- gbt.train(ytr, xtr, learning_rate = 0.1) gbt.complexity(model, type="xgboost") gbt.complexity(model, type="lightgbm") ## See demo(topic="gbt-complexity", package="agtboost")