from hypex.dataset import Dataset, InfoRole, TreatmentRole, TargetRole, DefaultRole, FeatureRole
from hypex import Matching

data = Dataset(
    roles={
        "user_id": InfoRole(int),  # InfoRole для ID
        "treat": TreatmentRole(int),  # TreatmentRole для определения группы (контроль/таргет)
        "post_spends": TargetRole(float)  # TargetRole для целевой переменной
    },
    data="data.csv",
    default_role=FeatureRole(),  # Остальные столбцы - обычные фичи
)

from hypex.dataset import Dataset, InfoRole, TreatmentRole, TargetRole, DefaultRole, FeatureRole
from hypex import Matching

# Описываем датасет

data = Dataset(
    roles={
        "user_id": InfoRole(int),  # InfoRole для ID
        "treat": TreatmentRole(int),  # TreatmentRole для определения групп (контроль/таргет)
        "post_spends": TargetRole(float)  # TargetRole для целевой переменной
    },
    data="data.csv",
    default_role=FeatureRole(),  # Остальные столбцы - фичи
)

# Запуск Matching

test = Matching()  # Классический Matching (Maha distance + full metrics)
test = Matching(metric="att")  # Только ATT
test = Matching(distance="l2")  # Выбор метрики

# Получение результатов
result = test.execute(data)
result.resume  # Краткое резюме 
result.full_data  # Ранее df_matched. Wide df с парами 
result.indexes  # Только индексы пар