initialize attgt

Parameters:¶

data: DataFrame¶

df = df.set_index(['entity', 'time'])

where df is the dataframe to use, ‘entity’ should be replaced with the name of the entity column and ‘time’ should be replaced with the name of the time column.

cohort_column : str¶

cohort name

base_period: 'varying' | 'universal' = 'varying'¶

• "universal"

• "varying"

anticipation: int = 0¶

The number of time periods before participating in the treatment where units can anticipate participating in the treatment, and therefore it can affect their untreated potential outcomes

dosage_column: str = None¶

The name of the column to be used in case of multi-valued treatment, used to calculate cohort-time-stratum ATT.

If stratum name is None, fit() will return cohort-time ATT.