vdblm commited on
Commit
4f2a8a8
·
verified ·
1 Parent(s): 5c60f36

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -6
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.5.1
24
- - numpy ≥ 1.26.4
25
- - tqdm ≥ 4.67.1
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