Oct, 2023

UncertaintyPlayground:一种用于快速和简化的不确定性估计的 Python 库

TL;DR介绍了 UncertaintyPlayground,这是一个建立在 PyTorch 和 GPyTorch 上的 Python 库,用于在监督学习任务中进行不确定性估计。该库通过稀疏和变分高斯过程回归 (SVGPRs) 用于正态分布输出以及混合密度网络 (MDN) 用于混合分布,提供了高斯和多模态结果分布的快速训练。此外,UncertaintyPlayground 可以使用不同的超参数进行模型训练,并可视化一个或多个实例的预测区间。通过使用张量操作,该库可以在 CPU 和 GPU 上进行训练,并提供了针对 PyTorch 的各种速度优化技术。该库包含每个模块的单元测试,并确保与 GitHub Workflows(在线集成)和 Tox(本地集成)进行多平台持续集成。最后,该代码用 Google 风格的文档字符串进行了文档化,并提供了使用 MkDocs 和 MkDocStrings 创建的文档网站。