gen_propensity_list: gen_propensity_list

In rzgross/uRbanmatching:

Description

Takes in elements needed for propensity function, checks input and builds a named list.

Usage

gen_propensity_list(
  propensity_function = propensity_score_xgb(),
  oos_propensity = FALSE,
  n_folds = NULL
)

Arguments

propensity_function	A function that accepts a list with four elements: x_train, x_test, y_train, y_test, and forms predictions using x_test (I guess y_test isn't used yet)
oos_propensity	Logical, whether to predict out of sample for the propensity score.
n_folds	Default NULL; how many folds if using out of sample propensity.

Value

Named list, same names as input params.

rzgross/uRbanmatching documentation built on Dec. 22, 2021, 8:20 p.m.