Python non_stratified_matching Examples

Example #1

File: test_ctmatching.py Project: MacHu-GWU/ctmatching-project

def test_independent_index_matching():
    control = np.random.random((30, 6))
    treatment = np.random.random((10, 6))
    nn_index = non_stratified_matching(control, treatment)
    (
        selected_control_index,
        selected_control_index_for_each_treatment,
    ) = independent_index_matching(nn_index, k=5)
    assert len(selected_control_index) > len(set(selected_control_index))

Example #2

File: test_ctmatching.py Project: MacHu-GWU/ctmatching-project

def test_non_stratified_matching():
    control = np.random.random((100, 6))
    treatment = np.random.random((10, 6))
    nn_index = non_stratified_matching(control, treatment)