如何将python列表输入Tensorflow

时间:2018-09-03 00:22:23

标签: python tensorflow

我已经处理了一些数据以在tensorflow中使用到两个python列表中。不幸的是,Tensorflow不支持将Python列表作为拟合模型的输入。下面是我当前实现的简短代码片段:

data = loadData()
pythonListInputData = data[0]
pythonListInputLabels = data[1]
model = prepModel() #note that this uses keras
model.fit(pythonListInputData, pythonListInputLabels) #Leaving out my configuration settings here for simplicities sake

这会导致错误:

ValueError: Please provide as model inputs either a single array or a list of arrays.

当pythonListInputData是列表列表或数组列表时,出现此错误。

我已经阅读了Tensorflow的教程和文档,但是正在努力寻找任何有用的信息来完成这项工作。

编辑:数据结构如下:

pythonListInputData = data[0]

是整数列表的列表,例如[[234,1,4],[245,2,5],[123,5,11],...]

我还尝试了一种如下构造的替代方法

pythonListInputData = []
for entry in data[0]:
    pythonListInputData.append(array.array('I', entry))

其中data [0]是上述格式。

pythonListInputLabels = data[1]

是整数列表,例如[1,4,2,...]

1 个答案:

答案 0 :(得分:0)

山姆!我在这里发布,以防答案对将来的人们有用。我们在评论中回答了这个问题。

问题是keras /张量流模型期望一个numpy数组 np.array(listName)

那应该可以解决您的问题!干杯!