Matching with multiple (> 2) treatment groups
Source:R/matching.R
morie_matching_multi_treatment.RdFor each non-reference treatment level, matches treated units to the reference group via the chosen binary matching method.
Usage
morie_matching_multi_treatment(
data,
treatment,
covariates,
reference_group = NULL,
method = "nearest_neighbor"
)Examples
# \donttest{
set.seed(1)
df <- data.frame(treat3 = sample(0:2, 200, TRUE),
x1 = rnorm(200), x2 = rnorm(200))
morie_matching_multi_treatment(df, "treat3", c("x1", "x2"))
#> Warning: Fewer control units than treated units; not all treated units will get a match.
#> $`0`
#> $matched_data
#> treat3 x1 x2 ._treat_binary
#> 1 0 1.511672283 1.113433184 1
#> 3 0 0.797916438 0.314516768 1
#> 4 1 -0.974002561 0.222218072 0
#> 5 0 0.689372698 -0.843615390 1
#> 8 1 -0.956891881 0.067972390 0
#> 9 1 -0.869782874 -0.201961742 0
#> 12 0 0.068511533 0.766065384 1
#> 13 0 -0.323750755 0.038928117 1
#> 14 0 -1.086503047 0.014646812 1
#> 15 1 -1.015928947 -0.186316968 0
#> 16 1 -0.767790185 1.400590827 0
#> 17 1 -1.119720061 0.018485570 0
#> 18 1 -0.448174237 0.249196012 0
#> 20 0 -1.180490683 -0.963233180 1
#> 22 0 -1.311420592 1.286922041 1
#> 23 0 -0.096524923 0.458125259 1
#> 24 0 2.369719908 -1.452028290 1
#> 25 0 0.890626476 0.077342277 1
#> 26 1 -0.252183161 0.559895273 0
#> 27 0 -0.865763755 -0.074946570 1
#> 28 0 0.582586000 0.782697698 1
#> 29 1 -0.012529347 -0.172668564 0
#> 30 1 -0.374854762 -1.051293778 0
#> 31 1 0.317885735 0.729451279 0
#> 32 0 -0.488805635 0.262665243 1
#> 34 0 1.680278205 1.041060298 1
#> 36 1 0.713240520 -1.629578287 0
#> 37 1 -0.542881937 0.121040232 0
#> 39 1 -0.348594685 -0.531043106 0
#> 41 1 1.883182542 -1.720650662 0
#> 42 0 -0.928971079 0.106320620 1
#> 44 1 -0.614950271 -0.301196983 0
#> 45 0 -0.947075792 0.976202525 1
#> 46 0 0.598975150 0.456008669 1
#> 48 1 -0.206189002 -1.133202206 0
#> 49 1 -0.574295414 -0.869460351 0
#> 52 1 -0.430879530 -1.001801335 0
#> 53 1 -0.592225373 -0.819868279 0
#> 54 1 0.981116160 -0.974552469 0
#> 55 1 0.532409357 0.604309409 0
#> 56 0 -0.090456124 0.548787588 1
#> 57 1 0.156490492 0.916432657 0
#> 58 1 -0.737311691 2.661566367 0
#> 59 1 -0.201341206 -0.180257065 0
#> 60 1 1.102176595 0.685014766 0
#> 61 0 -0.016748256 3.266414519 1
#> 64 1 -0.703694254 -0.972442936 0
#> 67 1 -1.064165163 -1.572372699 0
#> 70 0 -0.490832752 0.040427682 1
#> 71 0 -1.045717652 -0.390009563 1
#> 72 0 -0.896211264 -1.819222230 1
#> 73 0 1.269387164 0.659180710 1
#> 75 1 0.775634319 1.616626337 0
#> 77 0 -0.365401797 -0.286823883 1
#> 78 0 0.816556449 1.750321889 1
#> 79 1 -0.060634778 0.116413613 0
#> 80 0 -0.501378318 1.384253158 1
#> 81 0 0.926062725 0.574220913 1
#> 82 0 0.036937691 0.136490812 1
#> 83 0 -1.066200174 0.914215989 1
#> 85 1 1.495223444 -0.339880639 0
#> 86 0 1.172158547 0.606264572 1
#> 87 0 -1.457707210 1.341130311 1
#> 91 1 1.408563357 0.013864804 0
#> 92 1 -0.541760362 -1.105305910 0
#> 93 1 0.278664724 -0.025162639 0
#> 95 1 1.576158181 0.370059748 0
#> 96 1 -1.475547635 -0.380824536 0
#> 100 0 2.229262202 0.584077124 1
#> 101 1 -1.514497008 -0.722753124 0
#> 102 1 -0.061707422 -0.629164665 0
#> 103 0 -0.147270790 -1.816206055 1
#> 106 1 0.496578173 -1.427167613 0
#> 108 1 -0.260736309 -0.759530213 0
#> 109 0 -0.705928586 -2.278776142 1
#> 110 1 -0.161178506 -0.114050766 0
#> 111 0 0.501321828 2.351856666 1
#> 114 0 0.005641985 0.788970858 1
#> 115 1 -2.904899060 0.461465382 0
#> 116 0 -1.107164819 -0.438069134 1
#> 118 1 -0.976830350 -2.222946131 0
#> 121 0 -1.596718014 -0.987919139 1
#> 122 0 0.490967373 0.728488688 1
#> 123 1 0.421603365 -0.884684915 0
#> 125 1 1.034514324 -1.044375407 0
#> 126 0 0.081810310 -1.718136190 1
#> 127 0 -0.082523762 0.803757539 1
#> 128 0 0.606073431 -1.501787170 1
#> 129 1 -0.887420145 -0.145499071 0
#> 130 0 0.105421390 0.579458612 1
#> 132 1 0.550393358 1.893910787 0
#> 135 1 0.702116711 -0.556542168 0
#> 138 0 -0.589812790 -0.607131453 1
#> 139 0 -1.714502297 0.679362430 1
#> 140 1 -0.420997898 -0.093557640 0
#> 141 1 0.310141377 -0.490086292 0
#> 142 0 1.702570586 1.410659383 1
#> 143 1 -0.443384804 -0.224573790 0
#> 149 0 -0.298909313 -0.635865043 1
#> 151 0 0.951498470 -2.333136685 1
#> 153 1 0.616966476 0.316964024 0
#> 154 0 0.513493716 -0.707469988 1
#> 156 0 1.723894131 0.620152195 1
#> 157 1 -0.206144567 0.099903068 0
#> 158 0 -1.314195140 1.807703488 1
#> 159 1 0.063474096 -1.502429979 0
#> 160 1 -0.231977451 0.285640465 0
#> 161 1 0.635060328 0.845706964 0
#> 163 0 -1.808327579 -0.256854125 1
#> 164 1 -0.213885951 -0.055856043 0
#> 169 1 0.381411259 0.761544352 0
#> 171 0 -0.124434986 -0.626368228 1
#> 173 0 0.673928676 1.695271084 1
#> 174 1 1.956425263 -1.761226294 0
#> 175 1 -0.269041012 0.198013015 0
#> 176 1 -1.244551518 0.397349099 0
#> 178 0 0.097396655 2.560273389 1
#> 179 0 -0.238386950 1.257127712 1
#> 180 0 -0.411827957 -0.534537686 1
#> 183 0 -1.096236783 1.007199535 1
#> 184 0 0.308308749 0.719291823 1
#> 185 1 0.344795124 -0.604711661 0
#> 186 1 1.539648088 0.539054406 0
#> 187 1 -0.329514192 -0.076830886 0
#> 189 1 -0.479255587 -0.854907551 0
#> 190 0 -1.514886795 0.032637295 1
#> 192 0 -0.519536667 -0.982249076 1
#> 193 1 -0.834559033 0.004101957 0
#> 194 0 -0.756647609 -0.233427178 1
#> 196 0 1.572432917 1.549712963 1
#> 197 0 1.007364366 0.087496917 1
#> 200 0 0.222573831 -0.245622588 1
#>
#> $n_treated
#> [1] 66
#>
#> $n_matched_control
#> [1] 66
#>
#> $match_pairs
#> treated_idx control_idx distance
#> 1 1 95 0.353472091
#> 3 3 59 0.262115676
#> 5 5 52 0.104434193
#> 12 12 79 0.313687206
#> 13 13 44 0.169991630
#> 14 14 141 0.205142399
#> 20 20 92 0.051415668
#> 22 22 26 0.320032683
#> 23 23 29 0.299186487
#> 24 24 106 0.036385313
#> 25 25 9 0.178790065
#> 27 27 39 0.204567083
#> 28 28 18 0.281428901
#> 32 32 129 0.205263749
#> 34 34 153 0.373320814
#> 42 42 85 0.150704699
#> 45 45 115 0.296350530
#> 46 46 110 0.291921462
#> 56 56 140 0.315385649
#> 61 61 58 0.307521106
#> 70 70 96 0.226617318
#> 71 71 53 0.193673078
#> 72 72 41 0.118702758
#> 73 73 157 0.305202583
#> 77 77 108 0.223130400
#> 78 78 16 0.207899037
#> 80 80 60 0.292732769
#> 81 81 17 0.318203003
#> 82 82 143 0.184867920
#> 83 83 160 0.278796904
#> 86 86 193 0.339376290
#> 87 87 55 0.300727274
#> 100 100 93 0.341311321
#> 103 103 67 0.092863037
#> 109 109 36 0.346708183
#> 111 111 75 0.344175619
#> 114 114 4 0.295997018
#> 116 116 123 0.173973661
#> 121 121 48 0.033579029
#> 122 122 37 0.316835261
#> 126 126 174 0.027714483
#> 127 127 175 0.294189470
#> 128 128 159 0.014287327
#> 130 130 187 0.324737016
#> 138 138 54 0.135053470
#> 139 139 164 0.312573864
#> 142 142 31 0.361112020
#> 149 149 125 0.160802580
#> 151 151 118 0.002957623
#> 154 154 64 0.157947385
#> 156 156 8 0.332865459
#> 158 158 57 0.387896308
#> 163 163 101 0.215004422
#> 171 171 49 0.127723778
#> 173 173 161 0.406865776
#> 178 178 132 0.306671420
#> 179 179 186 0.297235285
#> 180 180 189 0.154788364
#> 183 183 176 0.295166698
#> 184 184 91 0.308656962
#> 190 190 135 0.224348362
#> 192 192 30 0.029257120
#> 194 194 102 0.171151282
#> 196 196 169 0.407220731
#> 197 197 15 0.182940535
#> 200 200 185 0.168399423
#>
#> $method
#> [1] "nearest_neighbor (rmorie native)"
#>
#> $details
#> $details$engine
#> [1] "native-greedy-1d"
#>
#> $details$caliper
#> NULL
#>
#> $details$replace
#> [1] FALSE
#>
#> $details$n_neighbors
#> [1] 1
#>
#> $details$alpha
#> [1] 0.05
#>
#> $details$propensity_logit_sd
#> [1] 0.5012909
#>
#> $details$treatment_level
#> [1] 0
#>
#> $details$reference_group
#> [1] 1
#>
#>
#> attr(,"class")
#> [1] "morie_match_result" "list"
#>
#> $`1`
#> $matched_data
#> [1] treat3 x1 x2 ._treat_binary
#> <0 rows> (or 0-length row.names)
#>
#> $n_treated
#> [1] 0
#>
#> $n_matched_control
#> [1] 0
#>
#> $match_pairs
#> [1] treated_idx control_idx distance
#> <0 rows> (or 0-length row.names)
#>
#> $method
#> [1] "nearest_neighbor (rmorie native)"
#>
#> $details
#> $details$engine
#> [1] "native-greedy-1d"
#>
#> $details$caliper
#> NULL
#>
#> $details$replace
#> [1] FALSE
#>
#> $details$n_neighbors
#> [1] 1
#>
#> $details$alpha
#> [1] 0.05
#>
#> $details$propensity_logit_sd
#> [1] 0
#>
#> $details$treatment_level
#> [1] 1
#>
#> $details$reference_group
#> [1] 1
#>
#>
#> attr(,"class")
#> [1] "morie_match_result" "list"
#>
#> $`2`
#> $matched_data
#> treat3 x1 x2 ._treat_binary
#> 2 2 -0.47569828 0.837394397 1
#> 4 1 -0.97400256 0.222218072 0
#> 6 2 -0.95583910 0.443805326 1
#> 7 2 -1.23170706 0.055797111 1
#> 8 1 -0.95689188 0.067972390 0
#> 9 1 -0.86978287 -0.201961742 0
#> 10 2 -0.91068068 -1.158015254 1
#> 11 2 0.74127631 -0.592743271 1
#> 15 1 -1.01592895 -0.186316968 0
#> 16 1 -0.76779018 1.400590827 0
#> 17 1 -1.11972006 0.018485570 0
#> 18 1 -0.44817424 0.249196012 0
#> 19 2 0.47173637 0.149220736 1
#> 21 2 1.47025700 -0.066466774 1
#> 26 1 -0.25218316 0.559895273 0
#> 29 1 -0.01252935 -0.172668564 0
#> 30 1 -0.37485476 -1.051293778 0
#> 31 1 0.31788574 0.729451279 0
#> 33 2 2.65865803 0.543657859 1
#> 35 2 0.77958401 0.197506154 1
#> 36 1 0.71324052 -1.629578287 0
#> 37 1 -0.54288194 0.121040232 0
#> 38 1 0.88577837 -1.637421952 0
#> 39 1 -0.34859468 -0.531043106 0
#> 40 2 -1.00805458 0.953679792 1
#> 41 1 1.88318254 -1.720650662 0
#> 43 2 -0.29419645 -0.608668497 1
#> 44 1 -0.61495027 -0.301196983 0
#> 47 2 -1.52361488 1.294408067 1
#> 49 1 -0.57429541 -0.869460351 0
#> 50 2 -1.39016604 -0.754970292 1
#> 51 2 -0.07041738 -0.129635350 1
#> 53 1 -0.59222537 -0.819868279 0
#> 54 1 0.98111616 -0.974552469 0
#> 55 1 0.53240936 0.604309409 0
#> 57 1 0.15649049 0.916432657 0
#> 58 1 -0.73731169 2.661566367 0
#> 59 1 -0.20134121 -0.180257065 0
#> 60 1 1.10217660 0.685014766 0
#> 62 2 0.16178863 0.560600460 1
#> 63 2 2.02476139 -0.069017297 1
#> 64 1 -0.70369425 -0.972442936 0
#> 65 2 0.96079238 -0.546586586 1
#> 66 2 1.79048505 -1.688692328 1
#> 67 1 -1.06416516 -1.572372699 0
#> 68 2 0.01763655 -0.404987158 1
#> 69 2 -0.38990863 0.319286418 1
#> 74 2 0.59384095 0.459621673 1
#> 75 1 0.77563432 1.616626337 0
#> 76 2 1.55737038 -1.856190495 1
#> 79 1 -0.06063478 0.116413613 0
#> 84 2 -0.23845635 -1.800826318 1
#> 85 1 1.49522344 -0.339880639 0
#> 88 2 0.09505623 0.767287288 1
#> 89 2 0.84766496 0.193725667 1
#> 90 2 -1.62436453 1.140566691 1
#> 91 1 1.40856336 0.013864804 0
#> 92 1 -0.54176036 -1.105305910 0
#> 93 1 0.27866472 -0.025162639 0
#> 94 2 -0.19397274 -0.163673338 1
#> 95 1 1.57615818 0.370059748 0
#> 96 1 -1.47554764 -0.380824536 0
#> 97 2 -0.14460821 0.652952369 1
#> 98 2 -0.95320315 2.061341806 1
#> 99 2 0.40654273 -1.796644937 1
#> 101 1 -1.51449701 -0.722753124 0
#> 102 1 -0.06170742 -0.629164665 0
#> 104 2 1.54159307 -0.259289098 1
#> 105 2 -0.98185567 0.334632273 1
#> 106 1 0.49657817 -1.427167613 0
#> 107 2 1.69694788 1.938628233 1
#> 108 1 -0.26073631 -0.759530213 0
#> 110 1 -0.16117851 -0.114050766 0
#> 112 2 -1.01353967 1.596320803 1
#> 113 2 1.61475224 1.277445511 1
#> 115 1 -2.90489906 0.461465382 0
#> 117 2 1.54756693 -1.507807009 1
#> 118 1 -0.97683035 -2.222946131 0
#> 119 2 -0.10150345 -1.178672442 1
#> 120 2 0.04265025 -1.782760224 1
#> 123 1 0.42160337 -0.884684915 0
#> 124 1 1.87390390 -1.538441417 0
#> 125 1 1.03451432 -1.044375407 0
#> 129 1 -0.88742015 -0.145499071 0
#> 131 2 0.35287447 1.201525576 1
#> 132 1 0.55039336 1.893910787 0
#> 133 2 -1.13433097 -1.760226343 1
#> 134 2 1.46235154 0.924546803 1
#> 135 1 0.70211671 -0.556542168 0
#> 136 2 2.50711115 -0.180584138 1
#> 137 2 -1.89002714 1.447440285 1
#> 140 1 -0.42099790 -0.093557640 0
#> 141 1 0.31014138 -0.490086292 0
#> 143 1 -0.44338480 -0.224573790 0
#> 144 2 -1.19859708 -0.212495490 1
#> 145 2 -0.30738091 0.696378474 1
#> 146 2 0.62105420 0.915182505 1
#> 147 2 0.18190219 -0.923374316 1
#> 148 2 1.31840093 1.146873263 1
#> 150 2 -1.64822174 -0.886443308 1
#> 152 2 -1.11312295 -0.145490805 1
#> 153 1 0.61696648 0.316964024 0
#> 155 2 0.36945910 1.242155558 1
#> 157 1 -0.20614457 0.099903068 0
#> 159 1 0.06347410 -1.502429979 0
#> 160 1 -0.23197745 0.285640465 0
#> 161 1 0.63506033 0.845706964 0
#> 162 2 1.63464429 -0.995342638 1
#> 164 1 -0.21388595 -0.055856043 0
#> 165 2 0.07036614 -0.445005240 1
#> 166 2 0.54970767 0.069695103 1
#> 167 2 -0.69682355 -0.154671706 1
#> 168 2 0.39056594 -0.831263459 1
#> 169 1 0.38141126 0.761544352 0
#> 170 2 -0.01237277 -0.576506933 1
#> 172 2 1.46674459 0.481335326 1
#> 174 1 1.95642526 -1.761226294 0
#> 175 1 -0.26904101 0.198013015 0
#> 176 1 -1.24455152 0.397349099 0
#> 177 2 -0.39570292 0.029225495 1
#> 181 2 -1.57721805 -0.625227429 1
#> 182 2 -0.79727610 0.913848687 1
#> 185 1 0.34479512 -0.604711661 0
#> 186 1 1.53964809 0.539054406 0
#> 187 1 -0.32951419 -0.076830886 0
#> 188 2 0.94838935 1.849919560 1
#> 189 1 -0.47925559 -0.854907551 0
#> 191 2 0.43453666 -1.025059481 1
#> 193 1 -0.83455903 0.004101957 0
#> 195 2 1.08950350 -0.498888219 1
#> 198 2 -0.27315799 1.318701131 1
#> 199 2 -1.30967954 -0.981224119 1
#>
#> $n_treated
#> [1] 66
#>
#> $n_matched_control
#> [1] 66
#>
#> $match_pairs
#> treated_idx control_idx distance
#> 2 2 29 2.155207e-01
#> 6 6 41 9.360715e-02
#> 7 7 108 5.489541e-02
#> 10 10 115 9.595969e-02
#> 11 11 176 9.421641e-02
#> 19 19 110 2.064664e-01
#> 21 21 26 1.525633e-01
#> 33 33 186 2.245569e-01
#> 35 35 157 2.263259e-01
#> 40 40 59 1.847699e-01
#> 43 43 38 7.827196e-02
#> 47 47 187 1.798510e-01
#> 50 50 92 6.240077e-02
#> 51 51 174 9.314518e-02
#> 62 62 18 2.165611e-01
#> 63 63 16 1.089284e-01
#> 65 65 125 1.370337e-01
#> 66 66 17 5.997524e-02
#> 68 68 123 6.566153e-02
#> 69 69 4 1.460687e-01
#> 74 74 93 2.106240e-01
#> 76 76 36 9.926130e-02
#> 84 84 96 1.861416e-01
#> 88 88 160 2.120323e-01
#> 89 89 175 2.213874e-01
#> 90 90 140 1.362080e-01
#> 94 94 193 7.660509e-02
#> 97 97 135 1.998222e-01
#> 98 98 55 1.478689e-01
#> 99 99 159 2.126726e-02
#> 104 104 79 2.049961e-01
#> 105 105 8 7.630464e-02
#> 107 107 132 2.422774e-01
#> 112 112 85 8.989127e-02
#> 113 113 75 6.394879e-02
#> 117 117 44 6.345328e-02
#> 119 119 49 3.085024e-05
#> 120 120 64 9.816338e-02
#> 131 131 31 1.508746e-01
#> 133 133 118 1.096368e-01
#> 134 134 95 1.463227e-01
#> 136 136 57 1.343711e-01
#> 137 137 54 1.657035e-01
#> 144 144 53 6.421556e-02
#> 145 145 164 2.106504e-01
#> 146 146 91 1.176929e-01
#> 147 147 106 9.081275e-02
#> 148 148 60 1.839027e-01
#> 150 150 30 2.036862e-01
#> 152 152 189 8.983727e-02
#> 155 155 161 6.787859e-02
#> 162 162 185 1.381509e-01
#> 165 165 102 8.247344e-02
#> 166 166 37 2.022340e-01
#> 167 167 9 4.890656e-02
#> 168 168 129 4.582301e-02
#> 170 170 39 5.319137e-02
#> 172 172 169 1.310249e-01
#> 177 177 143 8.687153e-02
#> 181 181 101 1.721979e-02
#> 182 182 141 2.071038e-01
#> 188 188 58 8.875848e-02
#> 191 191 15 3.358839e-02
#> 195 195 124 1.604496e-01
#> 198 198 153 1.278390e-01
#> 199 199 67 1.312325e-01
#>
#> $method
#> [1] "nearest_neighbor (rmorie native)"
#>
#> $details
#> $details$engine
#> [1] "native-greedy-1d"
#>
#> $details$caliper
#> NULL
#>
#> $details$replace
#> [1] FALSE
#>
#> $details$n_neighbors
#> [1] 1
#>
#> $details$alpha
#> [1] 0.05
#>
#> $details$propensity_logit_sd
#> [1] 0.3454421
#>
#> $details$treatment_level
#> [1] 2
#>
#> $details$reference_group
#> [1] 1
#>
#>
#> attr(,"class")
#> [1] "morie_match_result" "list"
#>
# }