我正在研究一个新冠肺炎图片分类问题.我有新冠肺炎和正常的CXRS图像.现在我想建立一个前馈神经网络.我写的这段代码一直给我一个错误,我不确定我到底做错了什么.
我的x_train.shape
是160,256,256,3
,x_test
是40,256,256,3
这就是我收到的错误
ValueError:
Input 0 of layer "sequential_4" is incompatible with the layer:
expected shape=(None, 196608), found shape=(None, 256, 256, 3)
附言:我是机器学习/深度学习的新手.
这是我的代码片段
import numpy as np
from keras.models import Sequential
from keras.layers import Flatten, Dense
# Define the architecture of the FNN model
model = Sequential()
model.add(Flatten(input_shape=(256,256,3)))
model.add(Dense(64, activation='relu'))
model.add(Dense(64, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
# Compile the FNN model
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
# Train the FNN model
model.fit(x_train, y_train, epochs=10, batch_size=32)