Stability and Oracle Inequalities for Optimal Transport Maps between General Distributions

This repository contains the supplementary code to reproduce the main experiments and figures from our NeurIPS 2025 paper. All scripts assume Python 3.8+ and use a self-contained output/ directory next to the code.

⸻

Project Structure

├── README.md                 # This file
├── requirements.txt          # Pip requirements
├── dataset.py                # Self-defined Dataset
├── utils.py                  # Help functions
├── network.py                # TNN definitions
└── sieve_sim-example.py      # Run an example of TNN experiment


⸻

Environment Setup

Tested on Python 3.11.11 (Ubuntu 22.04). Any Python 3.8+ interpreter should work.

Pip / virtualenv

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

⸻

Quickstart & Usage

All scripts infer the project root via Path(__file__).resolve().parent and write into output/.

Plots will be collected output/figures-TNN-d{d}/.


