zenflow

这个库实现了一个基于流的生成模型和作为FLAX模块实现的bijectors。支持条件流。

许可证

源代码在MIT许可证下发布。

安装

pip install zenflow

文档

目前没有在线文档，但库中有有用的docstrings。请使用docstrings并查看examples文件夹中的使用示例。

历史

该项目最初由PZFlow的John Franklin Crenshaw分叉，但大部分是重写的。PZFlow本身来自其他存储库，这些存储库在PZFlow文档中列出。我需要一个简单易懂、精简到基本功能的代码库，以便于我的使用场景。PZFlow和zenflow之间的区别

• zenflow使用通用的JAX数组进行数据输入和输出，而PZFlow强制使用Pandas。
• zenflow将所有可训练对象实现为FLAX模块，而PZFlow使用JAX原语。像FLAX一样，zenflow遵循清晰的函数式设计。
• PZflow支持在具有不确定性的数据点上训练，zenflow没有支持这一点。
• PZflow支持计算边缘后验密度，zenflow没有支持这一点。
• PZflow支持周期性数据，zenflow目前不支持这一点（我计划稍后重新引入）。
• PZflow支持的bijectors比zenflow多。