计算CNN中的假阳性率(FPR)和真阳性率(TPR)

时间:2019-04-07 14:49:11

标签: keras conv-neural-network

我正在设计用于对两种类型的图像进行分类的CNN,并且需要计算FPR和TPR。

在下面,您可以看到我的代码,但是我不知道如何基于此代码计算FPR和TPR。 您能否让我知道我该怎么做。 我知道要计算FPR和TPR,我可以使用以下代码

fpr, tpr, thresholds = metrics.roc_curve(y_test, y_predic)

y_predict可以由y_predic = model.predict(x_test)

计算

但是在我的代码中,我不知道该怎么做。 如果您能帮助我,我将非常感激。

num_classes = 1
batch_size = 512
epoch = 100

model = Sequential()

model.add(Conv2D(filters=8, kernel_size=(3, 3), padding='Valid', activation='relu', input_shape=(64, 64, 3)))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(rate=0.3))

model.add(Conv2D(filters=8, kernel_size=(3, 3), padding='Same', activation='relu'))
model.add(Dropout(rate=0.3))

model.add(Conv2D(filters=16, kernel_size=(3, 3), padding='Same',activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(rate=0.3))

model.add(Flatten())


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

model.add(Dense(units=num_classes, activation='sigmoid'))

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

train_datagen = ImageDataGenerator(rescale = 1./255,
                                   shear_range = 0.2,
                                   zoom_range = 0.2)


test_datagen = ImageDataGenerator(rescale = 1./255)

training_set = train_datagen.flow_from_directory('D:/Data/CNN/train',
                                                 target_size = (64, 64),
                                                 batch_size = batch_size,
                                                 class_mode = 'binary')

test_set = test_datagen.flow_from_directory('D:/Data/CNN/test',
                                            target_size = (64, 64),
                                            batch_size = batch_size,
                                            class_mode = 'binary', shuffle= True)


history = model.fit_generator(training_set,
                             steps_per_epoch = 4000//batch_size,
                             epochs = epoch,
                             verbose= 2,
                             validation_data = test_set,
                             validation_steps = 1000//batch_size) 

1 个答案:

答案 0 :(得分:1)

您几乎可以按原样使用keras.metrics。它同时具有FPR,TPR和SensitivityAtSpecificity

model.compile(optimizer='adam', loss=keras.losses.binary_crossentropy, metrics=['accuracy', tf.keras.metrics.SensitivityAtSpecificity(0.5)])

但是keras模型仅支持keras.metrics,因此如果不编写自定义指标,则不能使用tf.metrics

https://www.tensorflow.org/api_docs/python/tf/keras/metrics