#17 · 主分类: AI数据基础设施与存储

datasets

data dataset datasets jax machine-learning numpy tensorflow

TFDS是一个数据集集合,可直接用于TensorFlow、Jax等。

项目最后更新:08/21/26

GitHub Stars

4.6K

Forks数量

1.6K

贡献者数量

329

许可证

Apache-2.0

收录理由

很多团队真正耗时的环节往往不在模型本身,而是把数据从原始文件变成可训练的输入流。TensorFlow Datasets 把大量公开数据集统一到同一个加载接口下,几行代码就能拿到一个 tf.data.Dataset,下载、切分、预处理这些琐碎步骤都被封装好了,返回的数据结构直接兼容 TensorFlow,也能顺畅衔接 JAX 和 NumPy 生态。批处理、打乱、预取这些操作可以链式调用,不用自己再造轮子。它还顺带管理数据集的版本和引用信息,做可复现实验或者写论文要标注数据来源时能省下不少功夫。如果某个数据集不在内置列表里,项目文档也给出了扩展指引,照着步骤就能添加。不过最吸引人的还是开箱即用的体验和稳定的样本顺序,这让它成为机器学习数据准备环节里一个很顺手的基础工具。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目