简体   繁体   English

精度等于 0 CNN Python Keras

[英]Accuracy Equals 0 CNN Python Keras

I'm working on a binary classification problem.我正在研究二进制分类问题。 I was getting 69% accuracy at first, but kept running out of memory so I shrunk certain parameters, now it's coming up 0. Any idea whats going on?起初我得到了 69% 的准确率,但是一直用完 memory 所以我缩小了某些参数,现在它变成了 0。知道发生了什么吗?


model = Sequential()
from keras.layers import Dropout
model.add(Conv2D(96, kernel_size=11, padding="same", input_shape=(300, 300, 1), activation = 'relu'))
model.add(MaxPooling2D(pool_size=(3, 3), strides=(2, 2)))

model.add(Conv2D(128, kernel_size=3, padding="same", activation = 'relu'))
model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))

model.add(Conv2D(128, kernel_size=3, padding="same", activation = 'relu'))
model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))

from keras.layers.core import Activation

model.add(Flatten())
# model.add(Dense(units=1000, activation='relu'  ))
model.add(Dense(units= 300, activation='relu'))
model.add(Dropout(0.2))

model.add(Dense(1))
model.add(Activation("softmax"))

model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])


from keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(
    featurewise_center=True,
    rotation_range=90,
    fill_mode='nearest',
    validation_split = 0.2
    )

datagen.fit(train)

train_generator = datagen.flow(train, train_labels, batch_size=8)


# # fits the model on batches with real-time data augmentation:
history = model.fit_generator(generator=train_generator,
                    use_multiprocessing=True,
                    steps_per_epoch = len(train_generator) / 8,
                    epochs = 5,
                    workers=20)

Softmax should only be used if you have a multiclass classification problem.仅当您遇到多类分类问题时才应使用 Softmax。 You have a single output from your Dense layer, so you should use sigmoid.你有一个来自密集层的 output,所以你应该使用 sigmoid。

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM