/LSTM_predict_merger_history

Jason's toy model for predicting halo mass from merger tree in TNG100

Primary LanguagePythonMIT LicenseMIT

LSTM_predict_merger_history

Jason's toy model for predicting the halo mass in next snapshot from current data. The data I use is the merger tree data in TNG100. Simple 2-layer LSTM network using keras. I only train 1k samples and better results can be achieved using more samples. Future editions will focus on other halo properties.