/CNN_BCI2b

Primary LanguagePython

CNN_BCI2b

使用pytorch1.4.0简单实现了CNN模型对运动想象BCI2b数据的二分类

train.py文件用于模型训练

read_data.py用于数据的获取

predict.py是加载训练好的模型进行预测

display.py是可视化训练过程中的的准确率和损失值