Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
license:
|
| 3 |
tags:
|
| 4 |
- Causal-Inference
|
| 5 |
- CausalPFN
|
|
@@ -17,12 +17,13 @@ The model is based on a transformer architecture with uncertainty quantification
|
|
| 17 |
|
| 18 |
## Requirements
|
| 19 |
|
| 20 |
-
|
| 21 |
-
-
|
| 22 |
-
-
|
|
|
|
|
|
|
| 23 |
- tqdm
|
| 24 |
-
- faiss-cpu
|
| 25 |
-
- scikit-learn ≥ 1.5.2
|
| 26 |
- huggingface_hub
|
| 27 |
|
| 28 |
## Installation
|
|
@@ -84,4 +85,4 @@ If you use this model in your research, please cite:
|
|
| 84 |
|
| 85 |
## License
|
| 86 |
|
| 87 |
-
This model is licensed under
|
|
|
|
| 1 |
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
tags:
|
| 4 |
- Causal-Inference
|
| 5 |
- CausalPFN
|
|
|
|
| 17 |
|
| 18 |
## Requirements
|
| 19 |
|
| 20 |
+
|
| 21 |
+
- Python 3.10+
|
| 22 |
+
- PyTorch 2.3+
|
| 23 |
+
- NumPy
|
| 24 |
+
- scikit-learn
|
| 25 |
- tqdm
|
| 26 |
+
- faiss-cpu
|
|
|
|
| 27 |
- huggingface_hub
|
| 28 |
|
| 29 |
## Installation
|
|
|
|
| 85 |
|
| 86 |
## License
|
| 87 |
|
| 88 |
+
This model is licensed under [Apache-2.0](https://github.com/vdblm/CausalPFN/blob/main/LICENSE).
|