keras TypeError:' int'对象不可迭代

时间:2018-05-21 02:08:07

标签: python keras

我使用以下代码导入数据集。

dataset_csv = np.loadtxt('/home/user/Downloads/dataset.csv', delimiter=',')


x_train=dataset_csv[:3000000,0:3]
y_train=dataset_csv[:3000000,3]
x_test=dataset_csv[3000000:,0:3]
y_test=dataset_csv[3000000:,3]

总数据集是3193739。

每个数据集都有以下形式。 (value1,value2,value3,class)

我尝试通过如下设置输入层来插入数据。

model =Sequential()
model.add(Dense(128, input_shape=3, activation='relu'))

但是,发生了以下错误。

  

TypeError:' int'对象不可迭代

我想知道为什么你不能读取整数类型以及如何使数据集可读。

1 个答案:

答案 0 :(得分:1)

问题不在于读取数据,而在于Dense Class中的input_shape参数。

model.add(Dense(128, input_shape=3, activation='relu'))

输入形状应该是一个元组。所以,将其修改为以下

model.add(Dense(128, input_shape=(3,), activation='relu'))