# GeneralCostOT
General costs for neural optimal transport

Tested with ```Python 3.7.12 and PyTorch 1.11.0+cu102```

**Repository structure:**

```notebooks/twomoons_toy.ipynb``` - toy experiments on moons

```notebooks/gaussian_toy.ipynb``` - toy experiments on gaussians

```notebooks/dataset_transfer.ipynb``` - dataset transfer experiments with weak energy cost

```notebooks/dataset_transfer_no_z.ipynb``` - dataset transfer experiments with weak energy cost without input noise

```notebooks/train_classifier.ipynb``` - train a classifier to evaluate the accuracy of the dataset transfer

```notebooks/test_transport.ipynb``` - plotting the results and metrics (pre-trained classfiers are needed)
