与Keras(GPU)并行读取线

时间:2017-09-25 15:17:47

标签: machine-learning keras gpu

我想知道是否可以在Keras中与我的GPU并行读取文件中的几行。想象一下,我有一个包含50个功能(即50列)和10k行的文件,我想并行处理,这是使用Keras做到这一点最有效的方法吗?此外,我不想训练我的模型,因为我已经处于项目的部署阶段。

1 个答案:

答案 0 :(得分:1)

阅读整个文件,创建一个形状为(10k, 50)的批次并将其提供给模型:

result = model.predict(inputData)