conda create --name histograph python=3.11.11
conda activate histograph
pip install -r requirements.txt
pip install . 
cd GINPool/

# IMDB-B
python main.py --add_zero_attn=true --attn_mask_dropout=false --batch_size=64 --dataset=IMDBBINARY --device=0 --hidden_dim=64 --lr=0.0001 --mha_dropout=0.5 --num_head=1 --special_pooling_type=temporal --use_fc=true
# IMDB-M
python main.py --add_zero_attn=true --attn_mask_dropout=false --batch_size=128 --dataset=IMDBMULTI --device=0 --hidden_dim=16 --lr=0.005 --mha_dropout=0.5 --num_head=4 --special_pooling_type=temporal --use_fc=true
# MUTAG
python main.py --add_zero_attn=false --attn_mask_dropout=false --batch_size=32 --dataset=MUTAG --device=0 --hidden_dim=64 --lr=0.01 --mha_dropout=0.5 --num_head=4 --special_pooling_type=temporal --use_fc=true
# PROTEINS
python main.py --add_zero_attn=true --attn_mask_dropout=false --batch_size=128 --dataset=PROTEINS --device=0 --hidden_dim=16 --lr=0.005 --mha_dropout=0.5 --num_head=4 --special_pooling_type=temporal --use_fc=true
# PTC 
python main.py --add_zero_attn=false --alpha_type=fixed --attn_mask_dropout=true --batch_size=64 --dataset=PTC --device=0 --hidden_dim=64 --lr=0.005 --mha_dropout=0.1 --num_head=4 --skip_connection=true --special_pooling_type=temporal --use_fc=false --use_layer_norm=false --weight_decay=0.001
# RDT-B
python main.py --add_zero_attn=true --alpha_type=fixed --attn_mask_dropout=true --batch_size=16 --dataset=REDDITBINARY --device=0 --hidden_dim=64 --lr=0.005 --mha_dropout=0.5 --num_head=1 --skip_connection=true --special_pooling_type=temporal --use_fc=false --use_layer_norm=true --weight_decay=0
# NCI1
python main.py --add_zero_attn=true --alpha_type=fixed --attn_mask_dropout=true --batch_size=64 --dataset=NCI1 --device=0 --hidden_dim=16 --lr=0.0001 --mha_dropout=0.2 --num_head=4 --skip_connection=false --special_pooling_type=temporal --use_fc=true --use_layer_norm=false --weight_decay=0.01