Update README.md
Browse files
README.md
CHANGED
|
@@ -13,18 +13,17 @@ This repository contains the model weights for CausalPFN, a transformer-based in
|
|
| 13 |
|
| 14 |
CausalPFN is a pre-trained model for amortized causal effect estimation via in-context learning. It allows for accurate estimation of conditional average treatment effects (CATE) and average treatment effects (ATE) without requiring model retraining for each new dataset.
|
| 15 |
|
| 16 |
-
The model is based on a transformer architecture with
|
| 17 |
-
- Long-context in-context learning
|
| 18 |
-
- Uncertainty quantification and calibration
|
| 19 |
|
| 20 |
## Requirements
|
| 21 |
|
| 22 |
- Python ≥ 3.10
|
| 23 |
-
- torch ≥ 2.
|
| 24 |
-
- numpy ≥ 1.
|
| 25 |
-
- tqdm
|
| 26 |
- faiss-cpu ≥ 1.9.0
|
| 27 |
- scikit-learn ≥ 1.5.2
|
|
|
|
| 28 |
|
| 29 |
## Installation
|
| 30 |
|
|
|
|
| 13 |
|
| 14 |
CausalPFN is a pre-trained model for amortized causal effect estimation via in-context learning. It allows for accurate estimation of conditional average treatment effects (CATE) and average treatment effects (ATE) without requiring model retraining for each new dataset.
|
| 15 |
|
| 16 |
+
The model is based on a transformer architecture with uncertainty quantification and calibration.
|
|
|
|
|
|
|
| 17 |
|
| 18 |
## Requirements
|
| 19 |
|
| 20 |
- Python ≥ 3.10
|
| 21 |
+
- torch ≥ 2.0
|
| 22 |
+
- numpy ≥ 1.24
|
| 23 |
+
- tqdm
|
| 24 |
- faiss-cpu ≥ 1.9.0
|
| 25 |
- scikit-learn ≥ 1.5.2
|
| 26 |
+
- huggingface_hub
|
| 27 |
|
| 28 |
## Installation
|
| 29 |
|