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

In agtboost: Adaptive and Automatic Gradient Boosting Computations

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"

Details

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

Value

list with type hyperparameters.

Examples

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")

agtboost documentation built on Nov. 24, 2021, 1:16 a.m.