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)
# the reference level (the largest) must hold at least as many units as each other level
df <- data.frame(treat3 = sample(rep(0:2, c(90, 55, 55))),
x1 = rnorm(200), x2 = rnorm(200))
morie_matching_multi_treatment(df, "treat3", c("x1", "x2"))
#> $`1`
#> $matched_data
#> treat3 x1 x2 ._treat_binary
#> 1 0 -0.30097613 0.94658564 0
#> 3 1 -0.65209478 -0.35232231 1
#> 5 0 -1.91435943 0.73958923 0
#> 6 0 1.17658331 -1.06345742 0
#> 8 0 -0.46353040 -0.28949937 0
#> 9 0 -1.11592011 -2.26488936 0
#> 10 0 -0.75081900 -1.40885046 0
#> 11 1 2.08716655 0.91601933 1
#> 13 0 -1.28630053 0.80328322 0
#> 14 0 -1.64060553 1.88747446 0
#> 15 0 0.45018710 1.47388118 0
#> 17 0 -0.31806837 0.37996269 0
#> 18 1 -0.92936215 -0.19279843 1
#> 19 1 -1.48746031 1.57789179 1
#> 23 1 -1.38442685 -1.91890982 1
#> 26 0 -0.23864710 1.31400217 0
#> 29 0 -0.61924305 -0.16931833 0
#> 33 1 -0.14439960 -0.57254260 1
#> 38 1 -0.07715294 -0.06884093 1
#> 39 0 -0.33400084 -1.16766233 0
#> 42 0 2.07524501 -0.14587563 0
#> 43 0 1.02739244 -0.16391096 0
#> 44 1 1.20790840 1.76355200 1
#> 45 1 -1.23132342 0.76258651 1
#> 46 0 0.98389557 1.11143108 0
#> 48 1 -1.46725003 0.16434184 1
#> 51 1 1.46458731 -2.12936065 1
#> 54 0 -0.92610950 -0.81117015 0
#> 55 1 -0.17710396 1.32400432 1
#> 56 1 0.40201178 0.61563685 1
#> 58 1 0.83037317 0.30660486 1
#> 59 1 -1.20808279 -0.11015876 1
#> 60 1 -1.04798441 -0.92431277 1
#> 61 0 1.44115771 1.59291375 0
#> 62 0 -1.01584747 0.04501060 0
#> 68 1 -0.33090780 -0.39086782 1
#> 70 1 2.49766159 -0.37565742 1
#> 76 1 0.42069464 0.65253645 1
#> 77 0 -0.40024674 1.12477245 0
#> 79 1 0.98783827 -0.50808622 1
#> 81 0 -0.30874057 1.01775423 0
#> 82 1 -1.25328976 -0.25116459 1
#> 83 1 0.64224131 -1.42999345 1
#> 84 0 -0.04470914 1.70912103 0
#> 86 1 0.00213186 -0.71037115 1
#> 88 1 -0.34096858 -1.75946874 1
#> 90 1 1.80314191 1.61234680 1
#> 94 1 0.26317565 -0.68113139 1
#> 101 0 0.56082073 0.89200839 0
#> 102 0 -1.18645864 -0.02571507 0
#> 103 1 1.09677704 -0.64766045 1
#> 106 1 1.03410773 1.77261118 1
#> 107 0 0.22348041 -0.01825971 0
#> 109 1 1.16296456 0.20516290 1
#> 111 0 -0.54479074 -1.36611193 0
#> 112 1 -0.25567071 -0.42410226 1
#> 113 1 -0.16612104 0.23680366 1
#> 114 0 1.02046391 -2.34272312 0
#> 115 1 0.13622189 0.96169663 1
#> 116 1 0.40716760 -0.60442573 1
#> 117 1 -0.06965481 -0.75287728 1
#> 120 1 1.14622836 0.05633184 1
#> 121 0 -2.40309621 0.50936941 0
#> 123 0 0.37472441 -1.00436198 0
#> 125 0 0.95101281 -0.45303708 0
#> 126 0 -0.38923718 2.16536850 0
#> 127 1 -0.28433066 1.24574667 1
#> 130 1 0.27005490 0.27936078 1
#> 131 0 -0.42218401 -0.70590613 0
#> 133 0 -0.33103298 1.48021396 0
#> 136 1 0.39437917 -1.61887685 1
#> 138 1 2.64916688 0.44088937 1
#> 141 0 -2.28912398 0.36438459 0
#> 142 0 0.74100116 0.23349984 0
#> 143 1 -1.31624516 1.19395526 1
#> 145 1 0.39813016 -0.35729885 1
#> 148 1 -0.70123167 -0.36212377 1
#> 149 1 -0.58061430 2.35055433 1
#> 150 1 -1.00107218 2.44653138 1
#> 151 1 -0.66817861 -0.16670328 1
#> 152 0 0.94518495 -1.04366744 0
#> 153 1 0.43370215 -1.97293493 1
#> 155 0 -0.39011866 -1.09057358 0
#> 158 0 -1.42625734 0.11110643 0
#> 160 0 0.13444766 -1.10891000 0
#> 161 0 0.76559900 0.30756662 0
#> 162 0 0.95513668 -1.10689447 0
#> 163 0 -0.05056570 0.34765365 0
#> 164 0 -0.30581542 -0.87326454 0
#> 166 1 -1.04729815 -0.29686864 1
#> 168 1 -0.38363211 0.01129269 1
#> 169 0 1.65414530 0.99160104 0
#> 171 0 0.08296573 -1.37271127 0
#> 174 0 0.32300650 -1.11422235 0
#> 176 1 0.09907849 -0.93590256 1
#> 177 1 -0.45413691 -0.96723946 1
#> 180 0 1.06916146 0.23149613 0
#> 181 0 -0.48397493 -0.42237241 0
#> 182 0 -0.12101011 0.37411839 0
#> 183 1 -1.29414000 -0.36600577 1
#> 184 0 0.49431284 1.19010145 0
#> 187 0 0.81470273 -0.88484957 0
#> 188 0 -1.86978879 0.20800648 0
#> 190 0 0.45613560 -1.68452065 0
#> 191 0 -0.35340029 -0.14422656 0
#> 193 0 -0.86403595 0.68139992 0
#> 195 1 -0.32710101 -1.19231644 1
#> 196 1 -1.56908219 1.16922865 1
#> 198 1 1.36443493 -0.45177375 1
#> 199 1 -0.33428136 1.64202821 1
#>
#> $n_treated
#> [1] 55
#>
#> $n_matched_control
#> [1] 55
#>
#> $match_pairs
#> treated_idx control_idx distance
#> 3 3 102 5.869145e-03
#> 11 11 46 3.529211e-02
#> 18 18 164 7.636652e-03
#> 19 19 125 6.275896e-04
#> 23 23 9 5.851261e-03
#> 33 33 123 1.598806e-04
#> 38 38 6 2.721899e-03
#> 44 44 169 1.925512e-02
#> 45 45 13 2.605495e-04
#> 48 48 190 1.145395e-02
#> 51 51 131 7.646203e-03
#> 55 55 180 2.318028e-03
#> 56 56 43 1.235677e-02
#> 58 58 81 1.178967e-02
#> 59 59 171 8.729987e-03
#> 60 60 111 9.691587e-04
#> 68 68 29 9.740582e-04
#> 70 70 184 5.488496e-03
#> 76 76 77 1.044567e-02
#> 79 79 163 7.083083e-04
#> 82 82 39 7.062955e-03
#> 83 83 160 5.058767e-03
#> 86 86 62 4.736161e-03
#> 88 88 10 3.429182e-04
#> 90 90 61 1.562220e-02
#> 94 94 191 9.725483e-04
#> 103 103 182 5.054197e-05
#> 106 106 126 3.872468e-02
#> 109 109 133 1.103302e-03
#> 112 112 8 1.917310e-03
#> 113 113 17 7.644845e-04
#> 115 115 26 1.843387e-03
#> 116 116 152 5.670206e-04
#> 117 117 174 1.559159e-03
#> 120 120 161 3.280996e-03
#> 127 127 142 7.395159e-03
#> 130 130 14 3.204041e-04
#> 136 136 155 6.254480e-03
#> 138 138 15 3.920341e-02
#> 143 143 107 3.791994e-03
#> 145 145 193 9.306954e-04
#> 148 148 158 6.507277e-03
#> 149 149 84 9.302565e-03
#> 150 150 42 6.714219e-04
#> 151 151 181 4.900478e-03
#> 153 153 54 1.063947e-03
#> 166 166 188 8.929116e-03
#> 168 168 162 1.912601e-04
#> 176 176 5 5.878021e-04
#> 177 177 114 6.749888e-03
#> 183 183 121 2.619722e-03
#> 195 195 141 4.250651e-03
#> 196 196 187 2.710710e-03
#> 198 198 1 1.176893e-04
#> 199 199 101 1.179978e-05
#>
#> $method
#> [1] "nearest_neighbor (rmorie native)"
#>
#> $details
#> $details$engine
#> [1] "native-nearest (MatchIt algorithm)"
#>
#> $details$caliper
#> NULL
#>
#> $details$replace
#> [1] FALSE
#>
#> $details$n_neighbors
#> [1] 1
#>
#> $details$alpha
#> [1] 0.05
#>
#> $details$propensity_logit_sd
#> [1] 0.06747087
#>
#> $details$treatment_level
#> [1] 1
#>
#> $details$reference_group
#> [1] 0
#>
#>
#> attr(,"class")
#> [1] "morie_match_result" "list"
#>
#> $`2`
#> $matched_data
#> treat3 x1 x2 ._treat_binary
#> 2 2 -0.528279904 0.004398704 1
#> 4 2 -0.056896778 -0.529695509 1
#> 6 0 1.176583312 -1.063457415 0
#> 7 2 -1.664972436 0.246210844 1
#> 9 0 -1.115920105 -2.264889356 0
#> 12 2 0.017395620 -0.191278951 1
#> 13 0 -1.286300530 0.803283216 0
#> 14 0 -1.640605534 1.887474463 0
#> 15 0 0.450187101 1.473881181 0
#> 16 0 -0.018559833 0.677268492 0
#> 20 2 -1.075192297 0.596234109 1
#> 21 0 1.000028804 -1.173576941 0
#> 22 2 -0.621266695 -0.155642535 1
#> 25 2 0.425100377 -2.592327670 1
#> 28 2 0.886422651 -0.429978839 1
#> 29 0 -0.619243048 -0.169318332 0
#> 30 2 2.206102465 0.612218174 1
#> 31 0 -0.255027030 0.678340177 0
#> 32 2 -1.424494650 0.567951972 1
#> 34 2 0.207538339 -1.363291256 1
#> 35 2 2.307978399 -0.388722244 1
#> 36 0 0.105802368 0.277914132 0
#> 37 0 0.456998805 -0.823081122 0
#> 39 0 -0.334000842 -1.167662326 0
#> 40 2 -0.034726028 -0.008309014 1
#> 41 2 0.787639606 0.128855402 1
#> 42 0 2.075245009 -0.145875628 0
#> 43 0 1.027392439 -0.163910957 0
#> 46 0 0.983895570 1.111431081 0
#> 47 2 0.219924804 -0.923206953 1
#> 49 0 0.521022743 1.154825187 0
#> 50 2 -0.158754605 -0.056521425 1
#> 52 0 -0.766082000 0.344845762 0
#> 53 2 -0.430211754 -1.904955446 1
#> 57 2 -0.731748173 1.091668956 1
#> 62 0 -1.015847465 0.045010598 0
#> 63 2 0.411974712 -0.715128401 1
#> 64 2 -0.381076051 0.865223100 1
#> 65 0 0.409401840 1.074440958 0
#> 66 2 1.688873286 1.895654774 1
#> 67 2 1.586588433 -0.602997304 1
#> 71 0 0.667066167 -0.366630946 0
#> 72 2 0.541327336 -0.295677453 1
#> 73 2 -0.013399523 1.441820410 1
#> 74 2 0.510108423 -0.697538292 1
#> 75 0 -0.164375832 -0.388167506 0
#> 78 2 -1.370207878 -0.772110803 1
#> 80 2 1.519745025 0.523620590 1
#> 81 0 -0.308740569 1.017754227 0
#> 85 2 -1.733218407 1.435069572 1
#> 89 2 -1.156572363 0.569722972 1
#> 91 2 -0.331132036 -1.637280647 1
#> 92 0 -1.605513412 -0.779568513 0
#> 93 2 0.197193439 -0.641176934 1
#> 95 2 -0.985826700 -2.033285596 1
#> 98 2 0.570507636 -0.024997639 1
#> 99 0 -0.059723276 0.592984721 0
#> 100 0 -0.098178744 -0.198195421 0
#> 101 0 0.560820729 0.892008392 0
#> 102 0 -1.186458639 -0.025715071 0
#> 104 2 -0.005344028 0.646359415 1
#> 105 2 0.707310667 -0.433832740 1
#> 107 0 0.223480415 -0.018259711 0
#> 108 2 -0.878707613 0.852814994 1
#> 111 0 -0.544790740 -1.366111931 0
#> 118 0 -0.247664342 -1.555611593 0
#> 119 2 0.695550807 -1.453893738 1
#> 122 2 0.572739555 -2.097882960 1
#> 123 0 0.374724407 -1.004361979 0
#> 124 0 -0.425267722 0.535771722 0
#> 125 0 0.951012808 -0.453037085 0
#> 128 2 0.857409778 0.595498034 1
#> 129 0 1.719627299 0.004884450 0
#> 132 0 -1.189113295 0.628017153 0
#> 133 0 -0.331032979 1.480213960 0
#> 134 2 -0.939829327 1.083429910 1
#> 135 2 -0.258932583 -0.813244257 1
#> 137 2 -0.851857092 -0.109655699 1
#> 139 0 0.156011676 1.350993980 0
#> 142 0 0.741001157 0.233499835 0
#> 144 2 0.919803678 -0.027909972 1
#> 146 2 -0.407528579 -1.146814136 1
#> 147 0 1.324258630 -0.517420484 0
#> 154 2 1.005159218 0.514671633 1
#> 156 2 0.376370292 2.284659326 1
#> 157 2 0.244164924 -0.885617573 1
#> 160 0 0.134447661 -1.108909998 0
#> 161 0 0.765598999 0.307566624 0
#> 163 0 -0.050565701 0.347653649 0
#> 164 0 -0.305815420 -0.873264535 0
#> 165 2 0.893673702 0.077303123 1
#> 167 0 1.971337386 -1.183242240 0
#> 170 2 1.512212694 1.593967454 1
#> 171 0 0.082965734 -1.372711271 0
#> 172 2 0.567220915 -0.249610933 1
#> 173 0 -1.024548480 1.159424527 0
#> 174 0 0.323006503 -1.114222348 0
#> 178 2 -0.655781852 0.047488592 1
#> 180 0 1.069161461 0.231496128 0
#> 182 0 -0.121010111 0.374118395 0
#> 184 0 0.494312836 1.190101447 0
#> 187 0 0.814702731 -0.884849568 0
#> 188 0 -1.869788790 0.208006479 0
#> 189 0 0.482029504 -0.047730172 0
#> 190 0 0.456135603 -1.684520646 0
#> 192 2 0.170489471 1.180213666 1
#> 193 0 -0.864035954 0.681399923 0
#> 194 2 0.679230774 0.143247631 1
#> 197 2 -0.367450756 0.079201709 1
#> 200 2 0.732750042 -0.769592322 1
#>
#> $n_treated
#> [1] 55
#>
#> $n_matched_control
#> [1] 55
#>
#> $match_pairs
#> treated_idx control_idx distance
#> 2 2 124 5.426857e-04
#> 4 4 118 1.457285e-04
#> 7 7 188 2.906551e-02
#> 12 12 36 1.040611e-04
#> 20 20 173 8.031824e-03
#> 22 22 29 6.734057e-04
#> 25 25 46 1.962810e-02
#> 28 28 180 8.689786e-03
#> 30 30 42 1.594071e-03
#> 32 32 92 1.041790e-02
#> 34 34 184 2.800926e-02
#> 35 35 167 2.766252e-02
#> 40 40 75 8.623505e-03
#> 41 41 37 2.245395e-02
#> 47 47 65 2.704186e-02
#> 50 50 99 3.289330e-03
#> 53 53 39 6.116708e-03
#> 57 57 193 8.175603e-03
#> 63 63 49 3.535840e-02
#> 64 64 81 6.443905e-03
#> 66 66 6 6.000741e-03
#> 67 67 129 2.854995e-03
#> 72 72 160 3.747589e-02
#> 73 73 111 9.627569e-04
#> 74 74 189 2.198748e-02
#> 78 78 132 1.185258e-02
#> 80 80 147 1.324279e-04
#> 85 85 14 1.183910e-03
#> 89 89 13 2.549378e-02
#> 91 91 100 5.298125e-03
#> 93 93 15 2.092941e-02
#> 95 95 133 2.838617e-04
#> 98 98 171 3.464690e-02
#> 104 104 16 2.792689e-03
#> 105 105 142 1.338675e-02
#> 108 108 62 2.036539e-03
#> 119 119 187 1.880491e-03
#> 122 122 125 1.041424e-02
#> 128 128 123 2.700300e-02
#> 134 134 102 5.780588e-03
#> 135 135 164 5.243937e-03
#> 137 137 52 1.208854e-04
#> 144 144 43 1.955723e-02
#> 146 146 182 3.284814e-04
#> 154 154 190 2.028553e-02
#> 156 156 163 9.539937e-03
#> 157 157 107 2.687921e-02
#> 165 165 71 2.111883e-02
#> 170 170 21 7.514291e-04
#> 172 172 101 3.231650e-02
#> 178 178 9 4.105424e-03
#> 192 192 139 6.822367e-03
#> 194 194 174 1.781860e-02
#> 197 197 31 6.411005e-05
#> 200 200 161 2.477386e-02
#>
#> $method
#> [1] "nearest_neighbor (rmorie native)"
#>
#> $details
#> $details$engine
#> [1] "native-nearest (MatchIt algorithm)"
#>
#> $details$caliper
#> NULL
#>
#> $details$replace
#> [1] FALSE
#>
#> $details$n_neighbors
#> [1] 1
#>
#> $details$alpha
#> [1] 0.05
#>
#> $details$propensity_logit_sd
#> [1] 0.1536283
#>
#> $details$treatment_level
#> [1] 2
#>
#> $details$reference_group
#> [1] 0
#>
#>
#> attr(,"class")
#> [1] "morie_match_result" "list"
#>
# }