EQGraphNet is an graph deep learning model used for earthquake magnitude estimation.
The paper is available in
EQGraphNet is based on Pytorch and Pytorch Geometric
Firstly please create a virtual environment for yourself
conda create -n your-env-name python=3.11
Then, there are some Python packages need to be installed
conda install pytorch=2.1.0 torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
conda install pyg -c pyg
conda install pandas
conda install matplotlib
conda install h5py
pip install basemap
The Dataset used in our paper can be downloaded from https://github.com/smousavi05/STEAD. Before running, you should donwload and store the data file in the folder dataset like
After the preparation of Dataset, you can run the programs in the foloder run_mag_predict to test the performance :
python run_EQGraphNet.py