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.

Sign up for GitHub

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

Jump to bottom

add variational nearest neighbor GP #2026

Merged
merged 42 commits into from
Jul 19, 2022
Merged

add variational nearest neighbor GP #2026

merged 42 commits into from
Jul 19, 2022

Conversation

LuhuanWu
Copy link
Contributor

@LuhuanWu LuhuanWu commented Jun 2, 2022

Implements Variational Nearest Neighbor Gaussian Processes by Wu, Pleiss and Cunningham 2022. This implementation introduces a nearest neighbor utility that requires faiss or scikit-learn package.

LuhuanWu and others added 30 commits May 30, 2022 14:58
@LuhuanWu
add nearest neighbor jitter setting
c264e2a
@LuhuanWu
add nearest-neighbors-utils
3ae6888
@LuhuanWu
add nearest-neighbor-variational-strategy
202e815
add VNNGP illustration
528cdd6
@LuhuanWu
Merge branch 'cornellius-gp:master' into vnngp
2b4aeed
@LuhuanWu
add docstring for nn-jitter value
a1b0ff8
@LuhuanWu
[WIP] edit docstring for NNVariationalStrategy
8db2b99
@LuhuanWu
remove logger and verbose flag in utils/nearest_neighbors
4792a76
@LuhuanWu
change nearest neighbor function name
85b137e
@LuhuanWu
update utils/nearest-neighbors docstring and simplify nn functionality
5638622
@LuhuanWu
fix bug in nnutils
f52ba0d
@LuhuanWu
fix nnutils bug
fc29bef
@LuhuanWu
fix nnutils bug
e799970
update VNNGP notebook
d009d37
@LuhuanWu
update VNNGP notebook
49f0eff
@LuhuanWu
add vnngp to variational.rst
a01ab65
@LuhuanWu
add nearest neighbors to utils.rst
fe57d62
@LuhuanWu
add VNNGP notebook ti index.rst
9b9d379
@LuhuanWu
Merge branch 'vnngp' of github.com:LuhuanWu/gpytorch into vnngp
c4ba315
@LuhuanWu
add vnngp to pacakge init file
c778154
@LuhuanWu
add NNUtil to utils/init
6d1899b
@LuhuanWu
add batch support for nearest neighbor utils
bf339dd
@LuhuanWu
add test for nnutils
8187101
@LuhuanWu
adapt nn_util changes in nn_variaitonal_strategy
41d7d8b
@LuhuanWu
add sklearn support to nearest neighbor utils
3c420b6
@LuhuanWu
fix device in nearest neighbor utils
352498d
@LuhuanWu
[WIP] add batch mode support for stochastic-kl in nearest_neighbor_va…
8102997 
…riational_strategy
@LuhuanWu
[WIP] add test for nearest_neighbor_variational_strategy
cbc180d
@LuhuanWu
add batch support for nearest-neighbor strategy
db6bc77
@LuhuanWu
fix nearest neighbor device issue
3f797ea
@gpleiss gpleiss enabled auto-merge (squash) July 19, 2022 02:48
@gpleiss gpleiss merged commit 573b8fd into cornellius-gp:master Jul 19, 2022
@gpleiss gpleiss deleted the vnngp branch July 19, 2022 02:51
gpleiss added a commit that referenced this pull request Aug 7, 2022
@gpleiss
Run unittests with and without the faiss library.
409642c 
faiss is now an optional package for GPyTorch, to be used in conjunction with the
VNNGP model (#2026) and hopefully more nearest neighbor GP methods in the future.
This PR ensures that our test suite tests the functionality of GPyTorch with and
without the faiss library.
@gpleiss gpleiss mentioned this pull request Aug 7, 2022
Merged
gpleiss added a commit that referenced this pull request Aug 7, 2022
@gpleiss
Run unittests with and without the faiss library.
e726dcc 
faiss is now an optional package for GPyTorch, to be used in conjunction with the
VNNGP model (#2026) and hopefully more nearest neighbor GP methods in the future.
This PR ensures that our test suite tests the functionality of GPyTorch with and
without the faiss library.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@gpleiss gpleiss gpleiss approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@LuhuanWu @gpleiss