train_it: Generate Prediction Model

Description Usage Arguments Value

View source: R/train_it.R

Description

Function to develop prediction model based on user's preferences.

Usage

1
train_it(target, input, pred_model, sl_lib_internal = NULL, ...)

Arguments

target

A vector of target data.

input

A vector, matrix, or dataframe of input data.

pred_model

Prediction model algorithm.

  • 'sl': SuperLearner The required parameters:

    • sl_lib: a set of methods used for estimating target value (e.g., ("SL.xgboost","SL.earth","SL.gam","SL.ranger"))

sl_lib_internal

The internal library to be used by SuperLearner

...

Model related parameters should be provided.

Value

prediction model


CausalGPS documentation built on Feb. 7, 2022, 5:12 p.m.