GitHub - cornellius-gp/gpytorch: A highly efficient and modular implementation o...
source link: https://github.com/cornellius-gp/gpytorch
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
README.md
GPyTorch (Alpha Relase)
GPyTorch is a Gaussian Process library, implemented using PyTorch. It is designed for creating flexible and modular Gaussian Process models with ease, so that you don't have to be an expert to use GPs.
This package is currently under development, and is likely to change. Some things you can do right now:
- Simple GP regression (example here)
- Simple GP classification (example here)
- Multitask GP regression (example here)
- Scalable GP regression using kernel interpolation (example here)
- Scalable GP classification using kernel interpolation (example here)
- Deep kernel learning (example here)
- And (more!)
If you use GPyTorch, please cite the following paper:
Gardner, Jacob R., Geoff Pleiss, Ruihan Wu, Kilian Q. Weinberger, and Andrew Gordon Wilson. "Product Kernel Interpolation for Scalable Gaussian Processes." In AISTATS (2018).
@inproceedings{gardner2018product,
title={Product Kernel Interpolation for Scalable Gaussian Processes},
author={Gardner, Jacob R and Pleiss, Geoff and Wu, Ruihan and Weinberger, Kilian Q and Wilson, Andrew Gordon},
booktitle={AISTATS},
year={2018}
}
Installation
Global installation
The easiest way to install GPyTorch is by installing the dependencies we require, PyTorch >= 0.3.0
and libfftw3 > 3.3.6
(source) using conda, and then installing
GPyTorch using pip. This can be accomplished globally using one of the two sets of commands below depending on whether you want CUDA support.
For CUDA/GPU support, run:
conda install fftw cffi pytorch torchvision cuda80 -c conda-forge -c pytorch pip install git+https://github.com/cornellius-gp/gpytorch.git
If you do not have or do not wish to use CUDA, instead run:
conda install fftw cffi pytorch torchvision -c conda-forge -c pytorch pip install git+https://github.com/cornellius-gp/gpytorch.git
If you install libfftw3 from source, be sure to run configure
with --enable-shared
. To use packages globally but install GPyTorch as a user-only package, use pip install --user
above.
Installation in a conda environment
We also provide two conda environment files, environment.yml
and environment_cuda.yml
. As an example, to install GPyTorch in a conda environment with cuda support, run:
git clone git+https://github.com/cornellius-gp/gpytorch.git
conda create -f gpytorch/environment_cuda.yml
source activate gpytorch
pip install gpytorch/
Documentation
Still a work in progress. For now, please refer to the following example Jupyter notebooks.
Development
To run the unit tests:
python -m unittest
By default, the random seeds are locked down for some of the tests. If you want to run the tests without locking down the seed, run
UNLOCK_SEED=true python -m unittest
Please lint the code with flake8
.
pip install flake8 # if not already installed
flake8
Acknowledgements
Development of GPyTorch is supported by funding from the Bill and Melinda Gates Foundation.
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK