PO_dmlmt_w: This function calculates the weights impled by Double Machine...

In MCKnaus/dmlmt: Double Machine Learning for Multiple Treatments

Description

This function calculates the weights impled by Double Machine learning.

Usage

PO_dmlmt_w(x, t, y, p_mat, nm_list, cs_i = NULL, cl = NULL)

Arguments

x	Matrix of covariates (N x p matrix)
t	Matrix of binary treament indicators (N x # of treatments matrix) each column contains a binary indicator for each tratment
y	Vector of outcomes
p_mat	N x # of treatments matrix with (generalized) propensity scores
nm_list	List with length = # of treatments containing list with the selected variable names in the outcome regression for each treatment level
cs_i	If not NULL, boolean vector to indicate that observation is on support
cl	If not NULL, vector with cluster variables

Value

PO_dmlmt_w returns a list containing the Double Machine Learning Weights, the IPW weights, the OLS weights, and the adjustment weights.

MCKnaus/dmlmt documentation built on Dec. 4, 2020, 9:48 a.m.