https://gitlab.aicrowd.com/aicrowd/practice-challenges/aicrowd_ECTRSY_challenge/raw/master/baselines/ECTRSY_baseline.ipynb?1