reg_tune: Regression Tune

In daltoolbox: Leveraging Experiment Lines to Data Analytics

Regression Tune

Description

Creates an object for tuning regression models

Usage

reg_tune(base_model, folds = 10)

Arguments

base_model	base model for tuning
folds	number of folds for cross-validation

Value

returns a reg_tune object.

Examples

# preparing dataset for random sampling
data(Boston)
sr <- sample_random()
sr <- train_test(sr, Boston)
train <- sr$train
test <- sr$test

# hyper parameter setup
tune <- reg_tune(reg_mlp("medv"))
ranges <- list(size=c(3), decay=c(0.1,0.5))

# hyper parameter optimization
model <- fit(tune, train, ranges)

test_prediction <- predict(model, test)
test_predictand <- test[,"medv"]
test_eval <- evaluate(model, test_predictand, test_prediction)
test_eval$metrics

daltoolbox documentation built on Nov. 3, 2024, 9:06 a.m.