使用1DCNN写一个关于脑电信号二分类的代码
时间: 2023-03-26 22:01:30 浏览: 118
可以的,以下是一个使用1DCNN进行脑电信号二分类的代码示例:
```python
import numpy as np
import tensorflow as tf
from tensorflow.keras import layers
# 加载数据
X_train = np.load('X_train.npy')
y_train = np.load('y_train.npy')
X_test = np.load('X_test.npy')
y_test = np.load('y_test.npy')
# 构建模型
model = tf.keras.Sequential()
model.add(layers.Conv1D(filters=32, kernel_size=3, activation='relu', input_shape=(X_train.shape[1], 1)))
model.add(layers.MaxPooling1D(pool_size=2))
model.add(layers.Flatten())
model.add(layers.Dense(64, activation='relu'))
model.add(layers.Dense(1, activation='sigmoid'))
# 编译模型
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
# 训练模型
model.fit(X_train, y_train, epochs=10, batch_size=32, validation_data=(X_test, y_test))
# 评估模型
loss, accuracy = model.evaluate(X_test, y_test)
print('Test accuracy:', accuracy)
```
这个代码使用了一个包含32个卷积核的1D卷积层,每个卷积核的大小为3,激活函数为ReLU。接着使用了一个最大池化层和一个展平层,最后连接了两个全连接层,其中输出层使用了sigmoid激活函数。模型使用adam优化器和二元交叉熵损失函数进行训练,训练10个epoch,批量大小为32。最后评估模型的准确率。
阅读全文