TensorFlow I/O 0.14.0数据源扩展包
TensorFlow 的 IO 扩展包,tensorflow_io-0.14.0-cp36m-win_amd64.whl
,用起来还挺方便的,是你想在 Windows 上玩点花样数据源的时候。像 Kafka、Parquet、甚至 HDFS 这种,官方原版还真不一定支持,这个包就能帮上大忙。
Windows 环境的 TensorFlow要自定义数据源,一般比较折腾,要么自己搞扩展,要么到处找兼容的第三方库。tensorflow_io
就是为了这些事儿而生的,装好就能直接用,接口也和原生的 TensorFlow 差不多,迁移挺。
像你要在tf.data.Dataset
里读取 Parquet 文件,只要加一行tensorflow_io
的 import,用tfio.experimental.parquet
系列方法就行,代码简单,效率也不赖。别说在服务器上,开发时本地调试也能跑得飞快。
对了,要注意这个.whl
包只支持Python 3.6
和TensorFlow 2.1
Windows 64 位用的,别下错了平台或者 Python 版本,不然 pip 装不上。
如果你有点小项目用Kafka
或者文件系统
做数据输入,还想省点事,就可以先从tensorflow_io
开始试试,稳定性和兼容性都还不错。
16.13MB
文件大小:
评论区