OAE
├── README.md
├── requirement.txt
├─speed-test
│ ├── binary_tree.sh
│ ├── model.py
│ └── tree.py
├── test
│ ├── model.py
│ ├── test_oae-b.py
│ ├── test_oae-b.sh
│ ├── test-oae.py
│ └── test-oae.sh
└── train
├── train_oae
│ ├── dataset.py
│ ├── main.py
│ ├── model.py
│ └── train_oae.sh
└── train_oae-b
├── dataset.py
├── main.py
├── model.py
└── train_oae-b.sh
- create an python environment
conda create -n oae python==3.8
- install the packages
conda activate oae
pip install -r requirement.txt
OAE: run
sh OAE/train/train_oae/train_oae.sh
OAE-b: run
sh OAE/train/train_oae-b/train_oae-b.sh
OAE: test
sh OAE/test/test_oae.sh
OAE-b: run
sh OAE/test/test_oae-b.sh