propensity_score_xgb: propensity_score_xgb

In rzgross/uRbanmatching:

Description

Function factory to predict treatment using xgboost.

Usage

propensity_score_xgb(
  nrounds = 50,
  nthread = 1,
  params = list(eta = 0.1, max.depth = 3),
  ...
)

Arguments

nrounds	Training rounds for the xgb algorithm.
nthread	Number of threads to use for fitting, default 1.
params	List of params to pass to xgboost, most likely something like eta and max.depth
...	Additional xgboost params.

Value

returns a function that accepts train_test_list and this returns a vector of predictions for the test data

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