build_panel_embeddings

Guide & worked example · Source

build_panel_embeddings(
    X,
    y,
    unit_ids,
    time_ids,
    include_outcome=True,
    standardize=True,
)

Build unit-level embeddings by stacking panel moments over time.

Returns embeddings ordered by sorted unique unit IDs, along with the sorted unit and time levels used to create that ordering.