Are you planning on saving images of the waveform and classify it with a CNN?
This might work, if you see that these waveforms differ enough between classes.
However, I would also recommend to check, if creating spectra wouldn’t work better.