https://github.com/anindabitm/zindi_yield_prediction/blob/main/yield_nb.ipynb
Similar code using pytorch and fastai https://github.com/anindabitm/zindi_yield_prediction/blob/main/Pytorch_yield_nb.ipynb
I get 1.71 using LSTM model