Skip to content

Commit 567b029

Browse files
author
fabioferreira
committed
Port TensorFlow estimators/tests to PyTorch: updated README and added CI release to pypi
1 parent 7394564 commit 567b029

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

.github/workflows/ci.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ on:
44
push:
55
branches:
66
- main
7+
tags:
8+
- "v*"
79
pull_request:
810
branches:
911
- main

README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22

33
# Conditional Density Estimation (CDE)
44

5-
**Update:** Conditional Density Estimation now runs with PyTorch (with the help of cursor + GPT-5.1 Codex Mini). The legacy TensorFlow implementation lives in the `tensorflow` branch. All core estimators, runners, and examples are tested with the latest PyTorch release.
5+
**Update:** Conditional Density Estimation now runs with PyTorch (with the help of cursor + GPT-5.1 Codex Mini). The legacy TensorFlow implementation lives in the `tensorflow` branch. All core estimators, runners, and examples are tested with the latest PyTorch release. Logging is now done via wandb (offline/online mode & additional csv/pd dataframes outputs).
66

77
## Description
8-
Implementations of various methods for conditional density estimation
8+
Python/PyTorch implementations of various methods for conditional density estimation
99

1010
* **Parametric neural network based methods**
1111
* [Mixture Density Network (MDN)](https://publications.aston.ac.uk/id/eprint/373/)
@@ -44,6 +44,8 @@ After you activate the environment, install the local package in editable mode:
4444
```bash
4545
pip install --break-system-packages -e .
4646
```
47+
Prior to running experiments, set your Weights & Biases API key so wandb logging works (`export WANDB_API_KEY=<your-key>` or configure it via `wandb login` for online runs). You can store that key in a `.env` file (`WANDB_API_KEY=…`) and source it before launch. The tracking helpers also write CSV/PD outputs into `wandb/` when enabled.
48+
4749
If you already have a PyTorch environment, you can install the package with `pip install cde`; the runtime expects the usual scientific stack (`numpy`, `scipy`, `pandas`, `matplotlib`) and `ml_logger`.
4850
## Documentation and paper
4951
See the documentation [here](https://freelunchtheorem.github.io/Conditional_Density_Estimation). A paper on best practices and benchmarks on conditional density estimation with neural networks that makes extensive use of this library can be found [here](https://arxiv.org/abs/1903.00954).

0 commit comments

Comments
 (0)