search

Valentyn1997 / CausalTransformer

Code for the paper "Causal Transformer for Estimating Counterfactual Outcomes"
MIT License
95 stars 21 forks source link

Example for use with custom data #8

Closed JulianKlug closed 4 months ago

JulianKlug commented 6 months ago

Dear developing team, thanks for this great paper/project!

Currently it is quite complicated to guess the required dataset structure based only on the provided code.

Would it be possible to provide a small example or notebook on how to use this model with a custom dataset?

Valentyn1997 commented 5 months ago

Dear @JulianKlug. Great idea, I added a new markdown explaining the dataset structure here: https://github.com/Valentyn1997/CausalTransformer/blob/main/src/data/dataset-format.md