高斯过程在pytorch中的实现

gpytorch的Python项目详细描述


gpytorch(beta版)

Build statusDocumentation Status

forthebadge

新闻!

  • 测试版现在已经发布了!请注意,需要pytorch>;=1.1
  • 如果您需要安装alpha版本(我们建议您使用最新版本!),签出the alpha release

GPyTorch是一个使用PyTorch实现的高斯进程库GPHTARP是为了方便地创建可扩展的、灵活的和模块化的高斯过程模型而设计的。

内部,GPHARCH与许多现有的GP推理方法不同,通过使用预处理共轭梯度的现代数值线性代数技术来执行所有推理操作。实现一个可伸缩的GP方法非常简单,通过提供一个矩阵乘法例程,它通过我们的接口,或者通过组合我们已有的多个。这不仅使流行的可伸缩gp技术易于实现,而且与基于cholesky分解的解算器相比,gpu计算的利用率通常也有显著提高。

GPyPr火炬提供(1)显著的GPU加速(通过基于MVM的推理);(2)最新的可扩展性和灵活性的算法进展(SKI/KISS-GPstochastic Lanczos expansionsLOVESKIPstochastic variationaldeep kernel learning,…);(3)与深度学习框架的容易集成。

示例和教程

关于如何在gpytorch中构造各种模型,请参见我们的众多examples and tutorials。我们的ReadTheDocs页面here也提供了这些示例笔记本和GPyTorch的介绍

安装

要求

  • python=3.6
  • Pythorch=1.1个

n.b.gpytorch不会在pytorch 0.4.1或更早版本上运行。

首先确保使用here中的相应命令安装了PyTorch(>= 1.1

然后使用pip或conda安装gpytorch:

pip install gpytorch
conda install gpytorch -c gpytorch

要全局使用包,但将GPyTorch安装为仅限用户的包,请使用上面的pip install --user

最新(不稳定)版本

要获取最新(不稳定)版本,请运行

pip install git+https://github.com/cornellius-gp/gpytorch.git

引用我们的话

如果您使用gpytorch,请引用以下文章:

Gardner, Jacob R., Geoff Pleiss, David Bindel, Kilian Q. Weinberger, and Andrew Gordon Wilson. "GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration." In Advances in Neural Information Processing Systems (2018).

@inproceedings{gardner2018gpytorch,
  title={GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration},
  author={Gardner, Jacob R and Pleiss, Geoff and Bindel, David and Weinberger, Kilian Q and Wilson, Andrew Gordon},
  booktitle={Advances in Neural Information Processing Systems},
  year={2018}
}

文件

开发

运行单元测试:

python -m unittest

默认情况下,随机种子在某些测试中被锁定。 如果要在不锁定种子的情况下运行测试,请运行

UNLOCK_SEED=true python -m unittest

请用flake8对代码进行lint操作。

pip install flake8  # if not already installed
flake8

团队

gpytorch主要由以下人员维护:

Cornell LogoFacebook LogoUber AI Logo 我们要感谢我们的其他贡献者,包括(但不限于)大卫·阿尔布尔、Eytan Bakshy、大卫·埃里克森、贾里德·弗兰克、萨姆·斯坦顿、布拉姆·华莱士、Ke Alexander Wang、吴瑞汉。

致谢

gpytorch的开发由Bill and Melinda Gates FoundationNational Science FoundationSAP提供资金支持。

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java需要一种简单的方法来创建用于排序的comparator类   java getReadableDatabase和getWriteableDatabase无法解析   查找列表<Long>中与某个元素对应的所有索引的java方法   安卓将视图添加到ViewFlipper会导致java。StackOverflowerr语言   java根据它所包含的长“curTime”类字段将N的列表拆分为24(小时)   Android N中的Java8流API   自动生成Java策略文件的安全性   垃圾收集鼓励Java中的主要GC(但不是STW GC)   java如何检查UDP服务器上侦听的客户端数量   在前一台主机被Datastax Java驱动程序关闭后,Cassandra尝试重新连接到下一台主机   java如何使用Spring Boot创建部分代理   java是否有一个网站或资源可以完全比较EJB版本   java无需使用第三方库从gradle生成输出   继承由于这个多态性的基本示例中的语法有什么不同吗?(爪哇)   java字符串数组中的空字符串   java为什么CMS中的初始标记阶段是串行的   为什么Lucene有时与InChIKeys不匹配?   安卓通知Java应用程序数据库中的更改   java如何将单个json对象值解析为按钮   java打印堆栈将运行时错误跟踪到文件