https://www.kaggle.com/code/xiaowangiiiii/current-1th-jane-street-ae-mlp-xgb/notebook