请帮我写出基于keras搭建的CNN实现二分类并且输出准确率、精确率、召回率以及F1值
时间: 2024-01-29 15:04:23 浏览: 73
CNN训练实例(基于keras)
当然可以帮您实现基于Keras搭建的CNN进行二分类,并输出准确率、精确率、召回率以及F1值。以下是一个示例代码:
```python
import numpy as np
import tensorflow as tf
from tensorflow import keras
from sklearn.metrics import classification_report
# 加载数据集(假设已经准备好了训练集和测试集)
(x_train, y_train), (x_test, y_test) = keras.datasets.cifar10.load_data()
x_train = x_train.astype("float32") / 255.0
x_test = x_test.astype("float32") / 255.0
# 数据预处理
num_classes = 2
y_train = keras.utils.to_categorical(y_train, num_classes)
y_test = keras.utils.to_categorical(y_test, num_classes)
# 构建CNN模型
model = keras.Sequential([
keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)),
keras.layers.MaxPooling2D((2, 2)),
keras.layers.Conv2D(64, (3, 3), activation='relu'),
keras.layers.MaxPooling2D((2, 2)),
keras.layers.Conv2D(64, (3, 3), activation='relu'),
keras.layers.Flatten(),
keras.layers.Dense(64, activation='relu'),
keras.layers.Dense(num_classes, activation='softmax')
])
# 编译模型
model.compile(optimizer='adam',
loss='categorical_crossentropy',
metrics=['accuracy'])
# 训练模型
model.fit(x_train, y_train, epochs=10, batch_size=64, verbose=1)
# 在测试集上进行预测
y_pred = np.argmax(model.predict(x_test), axis=-1)
y_test = np.argmax(y_test, axis=-1)
# 输出准确率、精确率、召回率以及F1值
classification_result = classification_report(y_test, y_pred)
print(classification_result)
```
您可以根据自己的数据集和需要进行适当的修改。这段代码使用了CIFAR-10数据集作为示例,您可以根据自己的数据集进行加载和预处理。训练过程中使用了10个epoch,您可以根据需要进行调整。运行完后,会输出准确率、精确率、召回率以及F1值的报告。
希望对您有帮助!如有其他问题,请随时提问。
阅读全文