前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Github 项目推荐 | TensorFlow 概率推理工具集 —— probability

Github 项目推荐 | TensorFlow 概率推理工具集 —— probability

作者头像
AI研习社
发布2018-03-16 09:02:51
1.9K0
发布2018-03-16 09:02:51
举报
文章被收录于专栏:AI研习社

Probability 是 TensorFlow 的概率推理工具集,它是集建模工具、推理算法、一些有用的模型和一般统计计算于一身的开发工具集合。利用 TensorFlow,Probability 可以将概率方法和深度网络、通过自动差分的基于梯度的推论、大数据集、通过硬件(比如 GPU)加速的模型和分布式计算结合起来。

该软件的主要内容包括以下几个部分:

  • 采样算法,例如,tfp.metropolis_hastings,tfp.hmc,tfp.monte_carlo。
  • 示例模型(tfp.examples):使用此包和tf.contrib.distributions中的工具在TensorFlow中实现常见概率模型。

该库中的接口可能随时会更改。

Github 地址:

https://github.com/tensorflow/probability

安装:

安装 Probability 最简单的方法是用 pip:

代码语言:javascript
复制
pip install --upgrade tensorflow-probability       # for Python 2.7
pip3 install --upgrade tensorflow-probability      # for Python 3.n
pip install --upgrade tensorflow-probability-gpu   # for Python 2.7 and GPU
pip3 install --upgrade tensorflow-probability-gpu  # for Python 3.n and GPU

TensorFlow Probability 目前不包含任何指定 GPU(GPU-specific)的代码。这些软件包之间的主要区别在于 tensorflow-probability-gpu 取决于启用 GPU 的 TensorFlow 版本。

开发者也可以从源代码安装, 这需要 Bazel(https://bazel.build/)构建系统。

代码语言:javascript
复制
git clone https://github.com/tensorflow/probability.git
cd probability
bazel build --config=opt :pip_pkg
./bazel-bin/pip_pkg /tmp/tensorflow_probability_pkg
pip install /tmp/tensorflow_probability_pkg/*.whl

示例:

通过示例来学习是最容易的, examples / 目录包含常见概率模型的参考实现,并演示了在 TensorFlow 中构建概率模型的惯用方法。开发者可以直接用命令行运行示例:

代码语言:javascript
复制
python -m tensorflow_probability.examples.weight_uncertainty.mnist_deep_nn

训练贝叶斯深度网络对MNIST数字进行分类,请参阅示例目录:

https://github.com/tensorflow/probability/tree/master/examples/

用法:

安装 tensorflow_probability 后,可通过以下方式访问函数:

代码语言:javascript
复制
import tensorflow_probability as tfp 
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2018-02-24,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 AI研习社 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档