我正在设计用于对两种类型的图像进行分类的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)
答案 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