README.md

EnsNet

Run The Code

python main.py

Experimental Report

Reports of experimental results in the paper are listed in the directory "records/".

Run the checkpoint of the experimental results in the paper：

python LoadModel.py

Environment

python3.8

pip install torch
pip install torch-geometric