brain_wave prediction with cnn