我有多个 .csv 文件,每个文件都代表一个产品。我正在使用 LSTM 将这些产品分类为好或坏。所有 .csv 都以(#files、time_steps、#features)的 3d 矩阵的形式组合在一起。我在 batch_size=128 中传递矩阵。
````below is the psuedo code.
model = Sequential()
model.add(LSTM(#cells(64), input_shape=(time_steps, #features)))
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimiser='adam',
metrics=['accuracy'])
model.fit(train_X, train_y, batch_size=128, epochs=100, validation_split=0.1)
问题是我在损失函数中得到了 NaN,准确度也是 0.000。
任何人都可以建议我在哪里看。提前致谢。