Imported from https://github.com/neuralmind-ai/electricity-theft-detection-with-self-attention