agtboost (version 0.9.3)

gbt.complexity: Return complexity of model in terms of hyperparameters.

Description

gbt.complexity creates a list of hyperparameters from a model

Usage

gbt.complexity(model, type)

Arguments

model

object or pointer to object of class ENSEMBLE

type

currently supports "xgboost" or "lightgbm"

Value

list with type hyperparameters.

Details

Returns the complexity of model in terms of hyperparameters associated to model type.

Examples

# NOT RUN {
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")
# }
# NOT RUN {
# }