EHGNN icon indicating copy to clipboard operation
EHGNN copied to clipboard

Request to add required packages

Open TangYiChing opened this issue 1 year ago • 1 comments

Hi,

There are some missing references for the provided scripts. Is it possible to list at least the following packages required for classification_TU.sh, classification_OGB.sh, and classification_ZINC.sh? Thanks.

$ sh ./scripts/classification_TU.sh 9 000
Traceback (most recent call last):
  File "main.py", line 34, in <module>
    main(work_type_parser.parse_known_args()[0])
  File "main.py", line 7, in main
    from trainers.classification_TU import Trainer
  File "/home/EHGNN/trainers/classification_TU.py", line 6, in <module>
    **f**rom transformers.optimization import get_cosine_schedule_with_warmup
ModuleNotFoundError: No module named 'transformers'****

$ sh ./scripts/classification_OGB.sh 9 000
Traceback (most recent call last):
  File "main.py", line 34, in <module>
    main(work_type_parser.parse_known_args()[0])
  File "main.py", line 11, in main
    from trainers.classification_OGB import Trainer
  File "/home/EHGNN/trainers/classification_OGB.py", line 5, in <module>
    from ogb.graphproppred import Evaluator
ModuleNotFoundError: No module named 'ogb'

$ sh ./scripts/reconstruction_ZINC.sh 9 000
Traceback (most recent call last):
  File "main.py", line 34, in <module>
    main(work_type_parser.parse_known_args()[0])
  File "main.py", line 15, in main
    from trainers.reconstruction_ZINC import Trainer
  File "/home/EHGNN/trainers/reconstruction_ZINC.py", line 6, in <module>
    from torch_geometric.utils import to_dense_adj
ModuleNotFoundError: No module named 'torch_geometric'

TangYiChing avatar Apr 09 '23 21:04 TangYiChing