It is a binary classifier with a dataset of (4917,50,136) as (batch, step, features).
However, the dataset is unbalanced, class 0 contains 1401 data, and the rest of the
