Remove amplitude as a feature to avoid removing very large waveforms

It seems like the following units are being misclassified due to their amplitude

<img width="1182" height="884" alt="Image" src="https://github.com/user-attachments/assets/7fd3fcce-84e5-46eb-b4ce-f977b4a7f3bf" />

<img width="1182" height="873" alt="Image" src="https://github.com/user-attachments/assets/ac9d2fdf-4de4-443f-bb24-d8554a20d61f" />