peraktong / LSTM_predict_merger_history

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

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

License:MIT License


Languages

Language:Python 100.0%