Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[REVIEW] Adding initial artifacts for umap paper reproducible benchmarks #299

Open
wants to merge 1 commit into
base: branch-0.14
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 34 additions & 0 deletions publications/umap_paper_benchmarks/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
Datasets are not included in this repository and need to be downloaded separately.

The necessary dependencies for reproducing the benchmarks have been captured in `conda` environment yaml files.

To install dependencies for cuml and UMAP-learn benchmarks:
```
conda env create --name cuml_umap_benchmarks -f conda/umap_paper_cuml_cuda10.2.yml
```

To install dependencies for GPUMAP benchmarks:
```
conda env create --name gpumap_benchmarks -f conda/umap_paper_gpumap_cuda10.0.yml
```

You can run the notebooks using jupyter lab:
```
conda activate <environment_name>
python -m ipykernel install --user
jupyter lab
```

# Datasets

- PEN Digits - uses sklearn.datasets.load_digits
- GoogleNews Word2Vec - Downloaded from https://code.google.com/archive/p/word2vec/ and loaded using Gensim library
- Fashion MNIST - Downloaded from https://github.com/zalandoresearch/fashion-mnist
- CIFAR-100 - Downloaded from https://www.cs.toronto.edu/~kriz/cifar.html
- Shuttle - Downloaded from https://archive.ics.uci.edu/ml/datasets/Statlog+(Shuttle)
- MNIST - Uses datasets submodule to download and load
- TASIC2018 - Data from : https://portal.brain-map.org/atlases-and-data/rnaseq (see dedicated notebook)
- scRNA - Dataset downloaded from https://cells.ucsc.edu/
- COIL-20 - Uses datasets submodule to download and load


Empty file.
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
channels:
- rapidsai-nightly
- nvidia
- conda-forge
- facebook
- pytorch
- defaults
dependencies:
- python=3.7
- cudatoolkit=10.2
- cudf
- dask-cuda
- dask-cudf
- cuml
- cugraph
- opencv
- scikit-image
- chainer
- scipy
- gensim
- ucx-py
- joblib
- matplotlib
- umap-learn
- numba
- ucx-proc=*=gpu
- scikit-learn
- cupy
- ipykernel
- jupyterlab
- pip
- pip:
- jupyter-server-proxy
- git+https://github.com/dask/dask.git
- git+https://github.com/dask/distributed.git
- git+https://github.com/overshiki/datasets.git
- wget
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
channels:
- rapidsai-nightly
- nvidia
- conda-forge
- facebook
- pytorch
- defaults
dependencies:
- python=3.7
- cudatoolkit=10.0
- cudf
- dask-cuda
- dask-cudf
- cuml
- cugraph
- opencv
- scikit-image
- chainer
- scipy
- ucx-py
- joblib
- matplotlib
- umap-learn
- numba
- ucx-proc=*=gpu
- scikit-learn
- cupy
- ipykernel
- jupyterlab
- pip
- pip:
- jupyter-server-proxy
- faiss==1.5.3
- faiss-gpu==1.5.3
- git+https://github.com/dask/dask.git
- git+https://github.com/dask/distributed.git
- git+https://github.com/p3732/gpumap.git
- git+https://github.com/overshiki/datasets.git
- wget
Empty file.
880 changes: 880 additions & 0 deletions publications/umap_paper_benchmarks/notebooks/benchmarks_to_csv.ipynb

Large diffs are not rendered by default.

Loading